python - 如何使用列表中的数据框值

Question

我有一个数据框，其中包含列中的值列表，如下所示:

我想提取所有包含值“adhd”的行。我使用的代码:

df[df["Label"] == "adhd"]

但它返回给我一个只有列名的空数据框。是因为我的数据是列表格式吗？有什么方法可以处理这个问题，以便我可以删除行、查看行并使用它们。我宁愿将它们保留为列表，因为许多行都有多个值，例如 ["adhd","depression"]。

Answer 1

如果你想将它们保存在列表中，你可以执行像你这样声明的操作:

df[df['label'].apply(lambda x: x ==['adhd'])]  # keep people that only have ADHD 
df[df['label'].apply(lambda x: 'adhd' in x)]  # keep all people that have ADHD 

不过，我建议对您的数据进行单热编码，即每个标签保留一列(例如，如果人患有 ADHD，则 ADHD 的一列取值 1，否则为 0)。

我认为最简单的方法是:

df['label'].apply(pd.Series)

然后您可以更轻松地处理您的数据。