如何使用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=True并分配给新变量。