如何使用自定义标头将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

我可以使用熊猫吗？还是必须手动循环浏览内容并写入文件？

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('numrows numcols note\n')
    df.to_csv(file, header=False, index=False)

Answer 2

首先在追加模式下编写自定义字符串，然后编写所有不包含列的数据：

file = 'file.csv'
pd.DataFrame(columns=['numrows numcols note']).to_csv(file, index=False)
df.to_csv(file, header=None, index=False, mode='a')

Answer 3

通过@Divyanshu Srivastava改善答案 ：

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

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

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