根据另一列的值从一列中复制值

Question

我想根据第四列中的值从另外两列填充一列中的值。

我有四列的熊猫数据框：A，B，C，d

df_copy = df.copy() 
for i, row in df.iterrows(): 
    if 'Test' in row.D: 
     df_copy.loc[i, 'A'] = row.B 
    elif 'Other' in row.D: 
     df_copy.loc[i, 'A'] = row.C 

这工作，但速度很慢。有没有更高效的方法？

Answer 1

您可以使用“布尔索引”这个，而不是遍历所有行：

df_copy.loc[df['D']=='Test', 'A'] = df['B'] 
df_copy.loc[df['D']=='Other', 'A'] = df['C'] 

如果您知道列d仅由这两个值，它可以更短：

df_copy['A'] = df['B'] 
df_copy.loc[df['D']=='Other', 'A'] = df['C'] 

如果你想有一样的in运算符来测试如果子在列，你可以这样做：

df['D'].str.contains('Other') 

成为布尔值而不是df['D']=='Other'