我想基於上一列添加一個新的 DataFrame 列，以便如果上一列元素與列表值匹配，則更改該值

Question

輸入 df

Index       col1
0     Img    
1     Fruit  
2     Img    
3     Ball    
4     Ball    
5     Fruit    
6     shirt    
7     Fruit 

將列表映射到輸入 df

list1 = ['Img_A_10', 'Fruit_A_100', 'Ball_B_120']

輸出 df

     col1      col22
0     Img    Img_A_10
1     Fruit  Fruit_A_100
2     Img    Img_A_10
3     Ball   Ball_B_120
4     Ball   Ball_B_120
5     Fruit  Fruit_A_100  
6     shirt  shirt         
7     Fruit  Fruit_A_100

Answer 1

嘗試這個，

df['col2'] = df.col1.map({k.split("_")[0]: k for k in list1}).fillna(df.col1)

或者

df['col2'] = df.col1.replace({k.split("_")[0]: k for k in list1})

---

df
Out[93]: 
    col1         col2
0    Img     Img_A_10
1  Fruit  Fruit_A_100
2    Img     Img_A_10
3   Ball   Ball_B_120
4   Ball   Ball_B_120
5  Fruit  Fruit_A_100
6  shirt        shirt
7  Fruit  Fruit_A_100

Answer 2

以防萬一拆分不匹配（例如： A_Fruit_100 ），您可以extract然后replace

s = pd.Series(list1)
d = dict(zip(s.str.extract('('+'|'.join(df['col1'])+')',expand=False),s))
df['col22'] = df['col1'].replace(d)

---

print(df)
        col1        col22
Index                    
0        Img     Img_A_10
1      Fruit  Fruit_A_100
2        Img     Img_A_10
3       Ball   Ball_B_120
4       Ball   Ball_B_120
5      Fruit  Fruit_A_100
6      shirt        shirt
7      Fruit  Fruit_A_100