python - 根据包含条件屏蔽 Pandas DataFrame

Question

我正在构建一些处理图像的机器学习代码。我在屏蔽正确数据时遇到问题。

所以我在 pandas DataFrame 中有两列，图像的文件名及其标签。问题是，碰巧可能有一个标签与没有图像对应，因为我标记了一些后来被丢弃的图像。我试图掩盖它们，并且肯定会做一些丑陋的事情来做到这一点，但由于我是 Python 新手，我希望得到一些建议。

df 是一个 pandas DataFrame，具有 filename 和 file_attributes 列

df.drop(df[df['filename'] in os.listdir(PATH)].index, inplace=True)

我希望输出删除不在目录中的文件，但我收到错误:

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

Answer 1

在 pandas 中处理此类情况时，必须使用函数 isin() :

df.drop(df[df['filename'].isin(os.listdir(PATH))].index, inplace=True)