Skip to content

Commit bc965c2

Browse files
authored
Merge pull request #217 from debnath-d/patch-1
Update nsfg.py to fix FutureWarning
2 parents c1068c2 + 30c1a38 commit bc965c2

File tree

1 file changed

+11
-6
lines changed

1 file changed

+11
-6
lines changed

code/nsfg.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -65,12 +65,17 @@ def CleanFemPreg(df):
6565

6666
# replace 'not ascertained', 'refused', 'don't know' with NaN
6767
na_vals = [97, 98, 99]
68-
df.birthwgt_lb.replace(na_vals, np.nan, inplace=True)
69-
df.birthwgt_oz.replace(na_vals, np.nan, inplace=True)
70-
df.hpagelb.replace(na_vals, np.nan, inplace=True)
71-
72-
df.babysex.replace([7, 9], np.nan, inplace=True)
73-
df.nbrnaliv.replace([9], np.nan, inplace=True)
68+
df.replace(
69+
to_replace={
70+
"birthwgt_lb": na_vals,
71+
"birthwgt_oz": na_vals,
72+
"hpagelb": na_vals,
73+
"babysex": [7, 9],
74+
"nbrnaliv": [9],
75+
},
76+
value=np.nan,
77+
inplace=True,
78+
)
7479

7580
# birthweight is stored in two columns, lbs and oz.
7681
# convert to a single column in lb

0 commit comments

Comments
 (0)