如何使用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 = pd.DataFrame(data=sqldata.collect(),columns=['name','age'])