python - 带有 2 行标题并导出到 csv 的 pandas 数据框

Question

我有一个数据框

df = pd.DataFrame(columns = ["AA", "BB", "CC"])
df.loc[0]= ["a", "b", "c1"]
df.loc[1]= ["a", "b", "c2"]
df.loc[2]= ["a", "b", "c3"]

我需要在标题中添加第二行

df.columns = pd.MultiIndex.from_tuples(zip(df.columns, ["DD", "EE", "FF"]))

我的df现在是

  AA BB  CC
  DD EE  FF
0  a  b  c1
1  a  b  c2
2  a  b  c3

但是当我将这个数据帧写入 csv 文件时

df.to_csv("test.csv", index = False)

我比预期多了一行

AA,BB,CC
DD,EE,FF
,,
a,b,c1
a,b,c2
a,b,c3

Answer 1

这是一个丑陋的 hack，但如果您需要立即工作 (tm)，您可以将其分为两部分:

>>> pd.DataFrame(df.columns.tolist()).T.to_csv("noblankrows.csv", mode="w", header=False, index=False)
>>> df.to_csv("noblankrows.csv", mode="a", header=False, index=False)
>>> !cat noblankrows.csv
AA,BB,CC
DD,EE,FF
a,b,c1
a,b,c2
a,b,c3