python 检查文件夹内容是否存在

Question

这段代码的目的是:

读取包含文件名列表列的 csv 文件

这是 csv 文件:

https://drive.google.com/open?id=0B5bJvxM9TZkhVGI5dkdLVzAyNTA

然后检查特定文件夹以检查文件是否存在

如果发现文件不在列表中，请将其删除

这是代码:

import pandas as pd
import os.path

data = pd.read_csv('data.csv')
names = data['title']
path = "C:\\Users\\Sayed\\Desktop\\Economic Data"

for file in os.listdir(path):
    os.path.exists(file)
    print(file)
    file = os.path.join(path, file)
    fileName = os.path.splitext(file)

    if fileName not in names:
        print('error')
        os.remove(file)

我修改了第一个代码，这是新代码，我没有错误，但只需删除目录中的所有文件

Answer 1

os.chdir 不返回任何内容，因此将结果分配给 path 意味着 path 有 None ，这会导致错误。

由于您使用的是 pandas，因此这里有一个使用 pd.Series.isin 加快速度的小技巧。

root = "C:\Users\Sayed\Desktop\Economic Data"
files = os.listdir(root)

for f in data.loc[~data['title'].isin(files), 'title'].tolist():
    try:
        os.remove(os.path.join(root, f))
    except OSError:
        pass

根据 EAFP 添加了 try- except 检查(因为我在这里没有进行 os.path.exists 检查)。或者，您可以使用 pd.Series.apply 添加基于存在的过滤器:

m = ~data['title'].isin(files) & data['title'].apply(os.path.exists)

for f in data.loc[m, 'title'].tolist():
    os.remove(os.path.join(root, f))