如何使用python编写case语句

Question

我有一个名为类别的列

脚本 ：

import pandas as pd

df = pd.DataFrame({'SALES': ['B1','S1','S2']

                   })
df.head()
print(df)

我需要更换

B1 = BELTS
S1 = SHIRTS
S2 = SHOES 

建议我怎么做

Answer 1

下面是一个示例，说明如何使用 dict 映射旧值和新值：

import pandas as pd
df = pd.DataFrame({'SALES': ['B1','S1','S2']})
df['SALES'].replace({'B1': 'BELTS', 'S1': 'SHIRTS', 'S2': 'SHOES'}, inplace=True)

df
Out[1]: 
    SALES
0   BELTS
1  SHIRTS
2   SHOES

如果您想保持原始 DataFrame 原样，则跳过 inplace inplace=True并分配给新变量。