从 Google Colab 获取图像文件名

Question

我目前正在 Google Colab 的 Google Drive 中读取一些图像，需要检查每个图像文件名是否与 pandas 数据帧中的值相对应。

我的代码现在看起来是这样的：

for img in os.listdir(path):
    if img in df['column']:
        add image to a folder
    else: 
        add image to other folder

似乎 Python 正在检查实际的 图像 是否在数据框中，而不是图像名称并将所有图像添加到另一个文件夹。在与数据帧进行比较时，如何确保我实际上是在查看图像 文件名 ？谢谢！

Answer 1

_ in df[]

检查索引，而不是值。

尝试：

if img in df.values