python - Pandas 在读取 CSV 时重命名列

Question

我在一个文件夹中有多个文本文件，而不是输出以制作表格。我成功地将表格导出到 .csv 文件。问题是我想为每一列添加一个标题，并且我希望标题的名称是文本文件的名称。每个文件都是表中的一列数据。例如，第 1 列来自 textfile.1。我想在第 1 列中添加一个 header ，上面写着 "textfile.1"

这是我的工作代码:

import os

path = r'C:/path/to/file'
folders = os.listdir(path) #raw string

import pandas as pd

df = pd.DataFrame()
df_interim = pd.DataFrame()

for f in folders:
    df_interim = pd.read_csv(
        os.path.join(path,f ,),
        header=None
    )
    #concatenate the data into the original dataframe
    frames = [df, df_interim]
    df = pd.concat(frames, axis=1)
df.to_csv('outputfile',index=False)

Answer 1

修复你的代码，你只需要添加名称参数并将其设置为你的文件名:

df_interim = pd.read_csv(os.path.join(path, f),
                         header=None,
                         names=[f])

参见 here有关 read_csv 参数的更多信息。

---

这是一个更简单的方法，它删除了循环和 df_interim;

df_final = pd.concat([
    pd.read_csv(os.path.join(path, f), header=None, names=[f], squeeze=True)
    for f in folders
  ], axis=1
)