熊猫如果语句（excel相当于）

Question

我想在熊猫中创建一个简单的if语句。

的Excel版本如下：

= IF（E2 = “ABC”，C2，E2）

我卡在了如何基于字符串或部分字符串分配给它。

这是我的。

df['New Value'] = df['E'].map(lambda x: df['C'] if x == 'ABC' else df['E']] 

我知道我在这里犯了一个错误。

作为结果是在每个单元中的数据帧的整个值。

任何帮助将不胜感激！

Answer 1

使用np.where：

In [36]: 
df = pd.DataFrame({'A':np.random.randn(5), 'B':0, 'C':np.arange(5),'D':1, 'E':['asdsa','ABC','DEF','ABC','DAS']}) 
df 

Out[36]: 
      A B C D  E 
0 0.831728 0 0 1 asdsa 
1 0.734007 0 1 1 ABC 
2 -1.032752 0 2 1 DEF 
3 1.414198 0 3 1 ABC 
4 1.042621 0 4 1 DAS 

In [37]:  
df['New Value'] = np.where(df['E'] == 'ABC', df['C'], df['E']) 
df 

Out[37]: 
      A B C D  E New Value 
0 0.831728 0 0 1 asdsa  asdsa 
1 0.734007 0 1 1 ABC   1 
2 -1.032752 0 2 1 DEF  DEF 
3 1.414198 0 3 1 ABC   3 
4 1.042621 0 4 1 DAS  DAS 

的语法np.where是：

np.where(<condition>, True condition, False condition) 

所以当条件True它返回true;而当False其他条件。