无法使用map替换pandas dataframe列中的值

import pandas as pd

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

# 使用replace函数替换列中的值
replace_dict = {'apple': 'fruit', 'banana': 'fruit', 'orange': 'fruit',
                'red': 'color', 'yellow': 'color', 'orange': 'color'}
df = df.replace(replace_dict)

print(df)

       A      B
0  fruit  color
1  fruit  color
2  fruit  color

import pandas as pd

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

# 使用apply函数和lambda表达式替换列中的值
df['A'] = df['A'].apply(lambda x: 'fruit' if x in ['apple', 'banana', 'orange'] else x)
df['B'] = df['B'].apply(lambda x: 'color' if x in ['red', 'yellow', 'orange'] else x)

print(df)