如何使用pandas DataFrame append标题到_csv？

Question

sqldata = hiveSql.sql("""
        SELECT name,age from DB
""")

tabel_df = pd.DataFrame(data=sqldata.collect())
tabel_df.to_csv('resultRDD.csv')

如代码所示，我的CSV如下所示：

tom  16
jack 18

实际上，我需要如下CSV

name age
tom  16
jack 18

Answer 1

使用此选项：

tabel_df.to_csv('resultRDD.csv', header=['name','age'])

Answer 2

您还可以传入列名列表，如

tabel_df = pd.DataFrame(data=sqldata.collect(),columns=['name','age'])

Answer 3

只需创建列名列表：

table_df = pd.Dataframe(data=sqldata.collect())

#add these 2 lines
column_names = ['name','age']
table_df.columns = column_names

table_df.to_csv('resultRDD.csv')

就这样。完成了