使用键创建字典，因为文件名值是内容

Question

我有 4 个文件，我根据 csv 文件的名称创建了列表：

file_names['file1', 'file2', 'file3', 'file4']

在每个文件中我都有超过 6-7 列。我可以使用以下代码获取列名称：

for x in file_names:
    df = pd.read_csv(path + x)
    print(list(df.columns))

它工作正常，但是我如何创建（不是手动）带有 for 循环的字典，其中键file_names和值是list(df.columns)？

Answer 1

使用字典理解，nrows=0是为了更好地读取只读标头的性能：

d = {x: pd.read_csv(path + x, nrows=0).columns.tolist() for x in file_names}

或者通过以下方式循环读取标头readline：

d = {}
for x in file_names:
    with open(x) as f:
        d[x] = f.readline()