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