在目录中创建所有文件名及其文件扩展名的列表

Question

我正在尝试使用 pd.DataFrame 创建一个数据集来存储我目录中所有文件的文件名和文件扩展名。我最终想要有两个名为 Name 和 Extension 的变量。 name 变量将有一个文件名列表，而 extension 变量应该有一个文件类型，例如 xlsx 和 png。

我是 python 新手，只能做到这一点。这给了我一个文件名列表，但我不知道如何合并文件扩展名部分。有人可以帮忙吗？

List = pd.DataFrame()

path = 'C:/Users/documnets/'

filelist = []
filepath = []
# r=root, d=directories, f = files
for subdir, dirs, files in os.walk(path):
    for file in files:
        filelist.append(file)
        filename, file_extension = os.path.splitext('/path/to/somefile.xlsx')
        filepath.append(file_extension)


List = pd.DataFrame(flielist, filepath)

另外，对于这部分：os.path.splitext('/path/to/somefile.xlsx')，我可以保留括号中的内容还是应该用我的目录路径替换？

谢谢

Answer 1

你可以这样做：

import os
import pandas as pd
path = 'C:/Users/documnets/'

filename = []
fileext = []
for file in os.listdir(path):
    name, ext = file.split('.')
    filename.append(name)
    fileext.append(ext)

columns = ["Name", "Extension"]

data = [filename, fileext]
df = pd.DataFrame(data, columns).transpose()