如何在pandas中用字典键替换列值

import pandas as pd

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

# 创建一个字典，用于替换值
replace_dict = {'apple': 'fruit', 'orange': 'fruit', 'banana': 'fruit', 'red': 'color', 'yellow': 'color', 'green': 'color'}

# 使用replace函数替换列值
df['A'].replace(replace_dict, inplace=True)
df['B'].replace(replace_dict, inplace=True)

print(df)

       A      B
0  fruit  color
1  fruit  color
2  fruit  fruit
3  fruit  color