从不带掩码pandas的条件语句创建新列

Question

我正在寻找一种更好的方法来执行以下操作：

                 A      
 TRDNumber      
ALB2008081610   430     
ALB200808167    0       
ALB200808168    190     

使用条件语句基于另一列中的值创建新列

^{pr2}$

下面的代码可以工作，但我知道必须有更好的方法来实现这一点。在

mask = df['A'] == 0
df20 = df[mask] 
df20['B'] = 'x'
df20

mask2 = ((df.A != 0) & (df.B <= 200) )
df21 = df[mask2] 
df21['B'] = 'y'
df21

pieces = [df20,df21]
pd.concat(pieces)

Answer 1

您可以创建函数并将其apply添加到数据集：

>>> def foo(x):
...     if x['A'] == 0:
...         return 'x'
...     elif x['A'] < 200:
...         return 'y'
...     else:
...         return 'z'
... 
>>> df['B'] = df.apply(foo, axis=1)
>>> df
                 A  B
TRDNumber            
ALB2008081610  430  z
ALB200808167     0  x
ALB200808168   190  y