如果 A 列为 NULL，我需要将数据帧中的列 B 设置为 null

Question

给定以下数据帧：

A    B   C
NULL 1   2
3    4   3
5    3   1
NULL 2   6

如果 A 为空，我想用 null 替换 B 中的值：

A     B    C
NULL  NULL 2
3     4    3
5     3    1
NULL  NULL 6

有什么想法吗？

Answer 1

您可以使用

pd.DataFrame.loc。

请注意，任何具有NaN值的列都会变成float，因为NaN被认为是float。

df.loc[df['A'].isnull(), 'B'] = np.nan
print(df)
#      A    B  C
# 0  NaN  NaN  2
# 1  3.0  4.0  3
# 2  5.0  3.0  1
# 3  NaN  NaN  6