changehc: convert chained assignment to .loc

krivard · krivard · commit 9a9b423efc77 · 2021-01-25T16:35:04.000-05:00
diff --git a/changehc/delphi_changehc/load_data.py b/changehc/delphi_changehc/load_data.py
@@ -60,9 +60,7 @@ def load_chng_data(filepath, dropdate, base_geo,
         ]
 
     # counts between 1 and 3 are coded as "3 or less", we convert to 1
-    data[counts_col][
-        data[counts_col] == "3 or less"
-        ] = "1"
+    data.loc[data[counts_col] == "3 or less", counts_col] = "1"
     data[counts_col] = data[counts_col].astype(int)
 
     assert (

Original file line number	Diff line number	Diff line change
`@@ -60,9 +60,7 @@ def load_chng_data(filepath, dropdate, base_geo,`
`60`	`60`	`]`
`61`	`61`
`62`	`62`	`# counts between 1 and 3 are coded as "3 or less", we convert to 1`
`63`		`- data[counts_col][`
`64`		`- data[counts_col] == "3 or less"`
`65`		`- ] = "1"`
	`63`	`+ data.loc[data[counts_col] == "3 or less", counts_col] = "1"`
`66`	`64`	`data[counts_col] = data[counts_col].astype(int)`
`67`	`65`
`68`	`66`	`assert (`