Keiku
3/10/2017 - 1:48 AM

Impute some missing columns with pandas.

Impute some missing columns with pandas.

import pandas as pd

df = pd.DataFrame({'A':['A1', 'A2', 'A3'], 'B':[None, 'B2', None]})
df
# Out[51]:
#     A     B
# 0  A1  None
# 1  A2    B2
# 2  A3  None

# locでimpute
df.loc[(df['B'].isnull()), 'B'] = df['A']
df
# Out[53]:
#     A   B
# 0  A1  A1
# 1  A2  B2
# 2  A3  A3

# fillnaでimpute
df.A.fillna(df.B, inplace=True)
df
# Out[55]:
#     A   B
# 0  A1  A1
# 1  A2  B2
# 2  A3  A3