python - Python——读取文件，保存文件名并访问这些文件

Question

我建立

totalset = {} 

并从文件夹 A、B、C、D 中提取文件名

for file_name in (os.listdir(full_subdir_name)):
    full_file_name = os.path.join(full_subdir_name, file_name)

现在我想构建一个名为 namelist 的东西，其中包含所有文件名，我可以通过以下方式遍历文件夹中的所有文件：

for file in namelist[A]:
    blabla...

for file in namelist[B]:
    blabla.. .  

我应该怎么办？

Answer 1

您可能应该使用“列表理解”。这是一个如何做到这一点的示例：

lst_names = [os.path.join(full_subdir_name, n) for n in os.listdir(full_subdir_name)]

因此，您将构建完整的namelist类似这样的东西。首先，您应该设置一个名称列表。

# example using Windwows filenames
# note that we use "raw strings" with r'' so the backslash will not be weird
lst_names = [r'C:\Users\steveha\Desktop', r'C:\Users\steveha\Documents', r'C:\Users\steveha\Music']

# example using Mac or Linux filenames
lst_names = ['/home/steveha/Desktop', '/home/steveha/Documents', '/home/steveha/Music'

lst_names设置好姓名后，构建完整的namelist：

namelist = {}

for full_subdir_name in lst_names:
    namelist[full_subdir_name] = [os.path.join(full_subdir_name, n) for n in os.listdir(full_subdir_name)]

就我个人而言，我认为使用更短的变量名比阅读更容易full_subdir_name：

namelist = {}

for f in lst_names:
    namelist[f] = [os.path.join(f, n) for n in os.listdir(f)]