python - python Pandas 中的 if 语句

Question

我在 pandas 中有一个名为 family 的数据框，是我按照 sql 代码创建的。除了这一行之外，我已经完成了所有这些。 sql代码如下

SELECT IF(name Is Null, family.last_name, family.first_name) as family.NAME
from family

我尝试过以下方法

family['NAME']= np.where(family[family.name.isnull()],family.last_name, family.first_name)

但我没有取得任何进展，感谢任何帮助。数据框看起来像这样

name:    first_name:    last_name:
Peter    Peter          Smith
NaN      Phil           McGrath
Jack     Jack           Jones
NAN      Fred           Hogan 

我希望它给我一个看起来像这样的新专栏

NAME:
Peter
McGrath
Jack
Hogan

Answer 1

numpy.where(condition[, x, y])函数需要 bool 值作为 condition 参数:

condition : array_like, bool

When True, yield x, otherwise yield y.

所以你可以这样做:

np.where(family.name.isnull(),family.last_name, family.first_name)