python - 有条件地创建一个新列

Question

我有一个 pandas 数据框，y 有 2 列，A 和 B。A 列有一些空值，所以我想像这样创建一个 C 列，如果 A 有空值，则取 B 的值，否则取值A. 我正在使用此代码，但它不起作用：

if data['A'] is None:
    data['C'] = data['B']
else:
    data['C'] = data['A']

标签： pythonpandas

Answer 1

使用fillna-

data['C'] = data['A'].fillna(data['B'])