python - Pandas 通过比较两列来更新列

Question

这似乎是一个简单的问题，但我正在努力寻找解决方案。我有一个 pandas df，有两列(股票行情、目标)和一个名为 stock_symbols 的列表。简单来说，这就是我想要做的(注意，这不是我的代码):

For each value in df['ticker']:
    If df['ticker'] is in [ticker_list]:
        df['target'] = 1
    Else:
        df['target'] = 0

Answer 1

使用isin用于创建转换为 int 的 boolean mask (感谢 Michael Hoff ):

df['target'] = df['ticker'].isin(ticker_list).astype(int)

示例:

ticker_list = ['s','f']
df = pd.DataFrame({'ticker':['s','d','f']})
print (df)
  ticker
0      s
1      d
2      f

df['target'] = df['ticker'].isin(ticker_list).astype(int)
print (df)
  ticker  target
0      s       1
1      d       0
2      f       1