从目录中选择随机文件夹（不是文件）

Question

我要做的是让我的程序从目录中随机选择一个文件夹并读取其中的 slots.csv 文件。所有文件夹都有 slot.csv 文件，只是文件夹名称不同。我了解如何使用random.choice(os.listdir("C:\\"))，但我似乎无法让它与文件夹一起使用...

import os, random

random_centre = random.choice(os.listdir("Centres" + "\\" + postcode))
df = pandas.read_csv(r"Centres" + "\\" + postcode + "\\" + random_centre + "\\" + "slots.csv")

我不断收到同样的错误，

FileNotFoundError: [WinError 3] The system cannot find the path specified: 'Centres\\31400'

Answer 1

我没有 Pandas，所以这里有一个使用 Python 的 CSV 模块的模型。

我的第一条建议是使用os.path.join() 来组装您的路径。即使我在 Mac 上，此代码也应该可以在您的 Windows 机器上运行，因为join() 方法在选择路径分隔符时会考虑操作系统。这可能不是您的问题，但它会让您的代码更简洁。

至于随机问题，您的代码看起来应该可以工作：

import csv
import os
import random

postcode = '12345'
base_path = os.path.join('Centres', postcode)
random_centre = random.choice(os.listdir(base_path))
csv_path = os.path.join(base_path, random_centre, 'slots.csv')

with open(csv_path, newline='') as f:
    reader = csv.reader(f)

    print(csv_path)
    print(list(reader))

这是我运行几次后的样子：

% ./main.py
Centres/12345/b/slots.csv
[['Col1', 'Col2'], ['Val1', 'Val2']]

% ./main.py
Centres/12345/a/slots.csv
[['Col1', 'Col2'], ['Val3', 'Val4']]

% ./main.py
Centres/12345/a/slots.csv
[['Col1', 'Col2'], ['Val3', 'Val4']]

% ./main.py
Centres/12345/b/slots.csv
[['Col1', 'Col2'], ['Val1', 'Val2']]

% ./main.py
Centres/12345/a/slots.csv
[['Col1', 'Col2'], ['Val3', 'Val4']]

% ./main.py
Centres/12345/c/slots.csv
[['Col1', 'Col2'], ['Val5', 'Val6']]