Skip to main content
 首页 » 编程设计

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

2025年05月04日15lexus

我有 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) ?

请您参考如下方法:

使用字典理解,nrows=0 是为了提高只读标题的性能:

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

或者通过readline循环读取header:

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