Python 对列执行多个条件

我发现自己在一个Dataframe列中循环执行条件，根据这些条件，下面的结果与该单元格相邻。有没有更有效的方法来执行此操作

例如，我有数据帧df：

  A          B
potato
orange
watermelon
lettuce


... etc.

我将执行以下代码：

for i in range(len(df['A']):
    if df['A'].iloc[i] == 'watermelon':
        df['B'].iloc[i] = 'red'
    elif df['A'].iloc[i] == 'lettuce':
        df['B'].iloc[i] = 'green'
    elif .... :
   

结果:

  A          B
potato     brown
orange     orange 
watermelon red
lettuce    green

---

与

水果：颜色

字典一起使用，而不是循环或

apply（）

：

水果\u color=dict（土豆=‘棕色’、西瓜=‘红色’、莴苣=‘绿色’）
df['B']=df['A']。替换（水果颜色）
#A B
#0土豆棕
#1个橙子
#2西瓜红
#3生菜绿

与

水果：颜色

字典一起使用，而不是循环或

应用（）

水果\u color=dict（土豆=‘棕色’、西瓜=‘红色’、莴苣=‘绿色’）
df['B']=df['A']。替换（水果颜色）
#A B
#0土豆棕
#1个橙子
#2西瓜红
#3生菜绿