如何使用自定义标头将 pandas.DataFrame 写入 csv 文件？

Question

我有一个数据框

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

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

numrows numcols note
1 2
3 4

我可以用熊猫来做到这一点，还是我必须手动遍历内容并写入文件？

Answer 1

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

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

另外，请参阅这篇文章。

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

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

Answer 2

改进@Divyanshu Srivastava 的答案：

这并不是很重要，但不需要保留打开的文件：

with open(file_path, 'w') as f:
     f.write('Custom String\n')

df.to_csv(file_path, header=False, mode="a")