Pandas数据操作:将列映射到一些预定值

import pandas as pd

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

# 创建映射字典
mapping = {'apple': 'fruit', 'banana': 'fruit', 'orange': 'fruit'}

# 使用map()函数将列映射到预定值
df['A_mapped'] = df['A'].map(mapping)

print(df)

        A A_mapped
0   apple    fruit
1  banana    fruit
2  orange    fruit
3   apple    fruit

import pandas as pd

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

# 使用replace()函数将列映射到预定值
df['A_mapped'] = df['A'].replace(['apple', 'banana', 'orange'], 'fruit')

print(df)

        A A_mapped
0   apple    fruit
1  banana    fruit
2  orange    fruit
3   apple    fruit