Pandas使用map函数替换值

import pandas as pd

data = {'A': ['apple', 'banana', 'orange'],
        'B': ['red', 'yellow', 'orange']}
df = pd.DataFrame(data)

color_map = {'red': '红色', 'yellow': '黄色', 'orange': '橙色'}
df['B'] = df['B'].map(color_map)
print(df)

        A   B
0   apple  红色
1  banana  黄色
2  orange  橙色

import pandas as pd

data = {'A': ['apple', 'banana', 'orange'],
        'B': ['red', 'yellow', 'orange']}
df = pd.DataFrame(data)

color_series = pd.Series(['红色', '黄色', '橙色'], index=['red', 'yellow', 'orange'])
df['B'] = df['B'].map(color_series)
print(df)

import pandas as pd

data = {'A': ['apple', 'banana', 'orange'],
        'B': ['red', 'yellow', 'orange']}
df = pd.DataFrame(data)

def color_mapping(color):
    if color == 'red':
        return '红色'
    elif color == 'yellow':
        return '黄色'
    elif color == 'orange':
        return '橙色'

df['B'] = df['B'].map(color_mapping)
print(df)