python Pandas 如果列表中列中的单词在新列中添加 Y 标志

Question

我无法理解 python pandas 中的 if 语句。

我有一个数据框 df

Category | Count
A           45756
B           5857
C           57876

我创建一个列表，然后将其用作层次结构

list_s = {'A':'Y',
'B':'Y'}
df['Flag'] = df['Category'].replace(list_s)

但是我明白了

Category | Count   | Flag 
A           45756     Y
B           5857      Y
C           57876     C

代替

Category | Count   | Flag 
A           45756     Y
B           5857      Y
C           57876     

谢谢!

Answer 1

将 map 与 lambda 函数一起使用，您可以在其中使用字典的 get 方法，您可以在其中传递默认值

list_s = {'A':'Y', 'B':'Y'}
df['Flag'] = df['Category'].map(lambda x: list_s.get(x, ''))

print(df)

  Category  Count Flag
0        A  45756    Y
1        B   5857    Y
2        C  57876