Python Dataframe NaN 行切片、填充和重新连接

Question

我有一个大数据框。列中的某些值是 NaN。我想根据其他列值用一些值填充它们。

数据：

df = 

                             A          B

2019-10-01 09:19:40     667.029710    10

2019-10-01 09:20:15     673.518030    20

2019-10-01 09:21:29     533.137144    30

2020-07-25 15:51:15            NaN    40

2020-07-25 17:20:20            NaN    50

2020-07-25 17:21:23            NaN    60

我想NaN根据 B 列值填写 A 列。我的代码：

sdf = df[df['A'].isnull()] # slice NaN and create a new dataframe

sdf['A'] = sdf['B']*sdf['B']

df = pd.concat([df,sdf])

一切正常。我觉得我的代码很长。有一行代码吗？

Answer 1

因为fillna我们可以做到df.A.fillna(df.B**2, inplace=True)