python - 如何使用自定义 header 将 pandas.DataFrame 写入 csv 文件？

Question

我有一个数据框

import pandas as pd
df = pd.DataFrame([[1, 2], [3, 4]], columns=['a', 'b'])

我想将 df 写入 csv 文件但不使用列 ['a', 'b']。第一行是我自定义的字符串，其余是df.values的内容。例如:

numrows numcols note
1 2
3 4

我可以使用 pandas 执行此操作，还是必须手动循环遍历内容并写入文件？

Answer 1

您可以先用第一行中的自定义文本创建一个 csv 文件，然后将数据框附加到它。

with open('file.csv', 'a') as file:
    file.write('Custom String\n')
    df.to_csv(file, header=False, index=False)

另请参阅 this发布。

所以，在你的情况下，只需使用这个

with open('file.csv', 'a') as file:
    file.write('numrows numcols note\n')
    df.to_csv(file, header=False, index=False)