python - 在包含列表的 Pandas 列中查找字符串

Question

我对 Pandas 还很陌生，我真的被一个项目难住了。

我有一个如下所示的 Pandas 数据框:

<表类="s-表"><头>身份证零售商<正文>ID_1["零售商 1", "零售商 2", "零售商 3"]ID_2["零售商 1", "零售商 2"]

我的目标是:

• 遍历所有可能的零售商列表。
• 为每个可能的零售商创建一个列
• 如果当前零售商在该行的零售商列表中，则应将其编码为 1
• 如果该零售商不在列表中，则应将其编码为 0

以零售商3为例:

<表类="s-表"><头>身份证零售商零售商 3<正文>ID_1["零售商 1", "零售商 2", "零售商 3"]1ID_2["零售商 1", "零售商 2"]0

希望这是有道理的。

Answer 1

这是一种方法:

import pandas as pd

df = pd.DataFrame({'ID': {0: 'ID_1', 1: 'ID_2'},
 'Retailers': {0: ["retailer 1", "retailer 2", "retailer 3"],
  1: ["retailer 1", "retailer 2"]},
 'retailer 3': {0: 1, 1: 0}})

# get unique retailers
retailers = set(df['Retailers'].sum())
# create variables
for r in retailers:
    df[r] = df['Retailers'].apply(lambda x: r in x).astype(int)