如何将DataFrame及其列名保存到文本文件？

有两种简单方式可以实现需求：

方法一：用Pandas自带的to_csv方法（更推荐） #

直接借助DataFrame的to_csv方法，指定分隔符为空格，不保存索引并保留表头即可：

import pandas as pd

DF = pd.DataFrame({'A': [1], 'B': [2]})
DF.to_csv('Test.txt', sep=' ', index=False, header=True)

执行后Test.txt的内容为：

A B
1 2

方法二：继续使用np.savetxt（手动拼接表头） #

如果坚持使用np.savetxt，需要先把列名和数据合并成一个二维数组再保存：

import pandas as pd
import numpy as np

DF = pd.DataFrame({'A': [1], 'B': [2]})
# 拼接列名与数据行
combined_data = np.vstack([DF.columns, DF.values])
# 写入文件
np.savetxt('Test.txt', combined_data, fmt='%s')

生成的文件同样会包含表头和对应数据。

内容的提问来源于stack exchange，提问作者TourEiffel