在数据框中添加标题，保存它，然后读取并显示结果，但得到的不是我期望的。

Question

我有一个简单数据框，首先读取并显示它。然后我给数据框添加了一个标题，并再次显示结果，但得到的并不是我期望的结果。标题在每一列中都被重复了。以下是代码：

# 读取一个简单数据框
csvpath = r'C:\Users	fuser\Documents\2023 Taxes	est1.csv'
df = pd.read_csv(csvpath)
display(df)

# 给数据框添加标题并显示结果
text = '  TITLE'
df.columns = pd.MultiIndex.from_product([[text], df.columns])
display(df)

# 保存带有标题的数据框，然后再读取并显示
save_path = r'C:\Users	fuser\Documents\2023 Taxes\saved.csv'
df.to_csv(save_path, index=False)
recovered_df = pd.read_csv(save_path)
display(recovered_df)

结果显示如下：

A   B   C
0   A1  B1  C1
1   A2  B2  C2
2   A3  B3  C3

TITLE
A   B   C
0   A1  B1  C1
1   A2  B2  C2
2   A3  B3  C3

  TITLE TITLE.1 TITLE.2
0   A     B     C
1   A1    B1    C1
2   A2    B2    C2
3   A3    B3    C3

我知道问题与多级索引有关，但不知道如何修复以使输出仅显示单一标题，就像第二个输出所示那样。

Maxim Egorushkin · Answer

你需要在使用https://pandas.pydata.org/docs/reference/api/pandas.read_csv.html" rel="nofollow noreferrer">pd.read_csv时指定标题行的数量：

recovered_df = pd.read_csv(save_path, header=[0, 1])  # 使用2行作为列标题

输出结果如下：

>>> recovered_df
    TITLE        
        A   B   C
0      A1  B1  C1
1      A2  B2  C2
2      A3  B3  C3

>>> recovered_df.columns
MultiIndex([('  TITLE', 'A'),
            ('  TITLE', 'B'),
            ('  TITLE', 'C')],
           )

Nenad Vracar · Answer

尝试将以下代码：

df.columns = pd.MultiIndex.from_product([[text], df.columns])

替换为：

df.columns = [text + col for col in df.columns]

后者通过将文本与列名拼接来修改列名，而前者创建了一个具有独立级别的MultiIndex结构。这并不是你想要实现的目标。

Ken Thomases · Answer

假设我们有以下数据框：

import pandas as pd

data = {'A': ['A1', 'A2', 'A3'],
        'B': ['B1', 'B2', 'B3'],
        'C': ['C1', 'C2', 'C3']}

df = pd.DataFrame(data)

数据框内容如下：

    A   B   C
0  A1  B1  C1
1  A2  B2  C2
2  A3  B3  C3

为了保存带有标题的数据框，请按照以下步骤操作（关键在于将标题作为第一行CSV数据进行保存）：

import pandas as pd
import os

data = {'A': ['A1', 'A2', 'A3'],
        'B': ['B1', 'B2', 'B3'],
        'C': ['C1', 'C2', 'C3']}

df = pd.DataFrame(data)

savePath = os.path.join(os.getcwd(), "dataframe.csv")
# 将带有列标题的标题化数据框保存在当前目录下
df.to_csv(savePath, index=False, header=True)

# 定义数据框的标题
title = '我的数据框标题'

# 将数据框保存到CSV文件中，其中标题作为第一行
with open(savePath, 'w', encoding='utf-8') as f:
    f.write(title + '
')
    df.to_csv(f, index=False, header=True, sep=',')

# 将CSV文件重新读入数据框并显示
df2 = pd.read_csv(savePath)
print(df2)

df2 显示如下：

      我的数据框标题
A  B                 C
A1 B1               C1
A2 B2               C2
A3 B3               C3