通过对包含另一列中字符串值的子集的字典进行映射来创建dataframe列。作为它的关键

import pandas as pd

# 创建一个包含字符串的DataFrame
df = pd.DataFrame({'fruit': ['apple', 'banana', 'orange', 'apple']})

# 创建一个字典，将字符串映射为对应的颜色
fruit_color = {'apple': 'red', 'banana': 'yellow', 'orange': 'orange'}

# 使用map函数将fruit列中的字符串映射为颜色
df['color'] = df['fruit'].map(fruit_color)

print(df)

    fruit   color
0   apple     red
1  banana  yellow
2  orange  orange
3   apple     red