Pandas根据多列中的值进行映射

import pandas as pd

# 创建一个示例DataFrame
data = {'A': ['apple', 'banana', 'orange'],
        'B': ['red', 'yellow', 'orange'],
        'C': ['fruit', 'fruit', 'fruit']}
df = pd.DataFrame(data)

# 创建一个映射字典
mapping = {'apple': 'fruit', 'banana': 'fruit', 'orange': 'fruit',
           'red': 'color', 'yellow': 'color', 'orange': 'color',
           'fruit': 'category'}

# 使用map函数进行映射
df['D'] = df['A'].map(mapping)
df['E'] = df['B'].map(mapping)
df['F'] = df['C'].map(mapping)

print(df)

        A       B      C       D      E        F
0   apple     red  fruit   fruit  color  category
1  banana  yellow  fruit   fruit  color  category
2  orange  orange  fruit   fruit  color  category