python - 遍历目录路径

Question

我正在尝试列出所有目录(/home/目录下一级)及其内容。此代码有效，但是当我将字典转换为 pandas 数据框时，目录名与文件名不匹配。

mypath='/home/'
from os import walk

myd=dict()
for (dirpath, dirnames, filenames) in walk(mypath):
    for i in dirnames:
        for (dirpath1, dirnames1, filenames1) in walk(i):
            myd[i]=','.join(filenames1)


import pandas as pd
df=pd.DataFrame(myd , index=[0]).T
df.columns=['files']

pd.set_option('max_colwidth', 800)
df

有没有更好的方法来构建包含目录及其文件内容的 2 列数据框？

Answer 1

我不确定您的最终结果应该是什么样子，但是 os.walk为你做完整的递归!无需在第二个循环中遍历 dirnames:

import os

mypath = '/home/'

myd = {}
for (here, dirs, files) in os.walk(mypath):
    for file in files:
        myd[here] = '.'.join(files)

print(myd)

这是python 3代码； python 2 file 是关键字，不应用作变量名...

更新

如果您只需要在输入目录下一级，则无需walk:

myd = {}
for name in os.listdir(mypath):
    subdir = os.path.join(mypath, name)
    if not os.path.isdir(subdir):
        continue
    myd[name] = '.'.join(os.listdir(subdir))

print(myd)