python-3.x - 将逗号分隔的字符串转换为 pandas 中的列表

Question

我有一个有两列的数据框。每列都有逗号分隔的字符串。我正在尝试将此字符串转换为列表，因此我可以为以下数据集的每一行减去两个列表:a_b = list(set(a) - set(b))。

  Col1      Col2
  a,b,c,f   d,f,g
  d,g       w,a,d
  f,g,h     f,g,h

我尝试将项目转换为如下列表

df1['Col1']tolist()

但是好像不行。谢谢。

Answer 1

您可以使用str.split将逗号分隔的字符串拆分为 list。您还可以使用 apply(set) 来实现您的特定目的 IIUC:

(df['Col1'].str.split(',').apply(set) - df['Col2'].str.split(',').apply(set)).tolist()

[输出]

[{'a', 'b', 'c'}, {'g'}, set()]