首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何使用pandas DataFrame to_csv添加标题?

如何使用pandas DataFrame to_csv添加标题?
EN

Stack Overflow用户
提问于 2019-12-21 11:31:32
回答 4查看 167关注 0票数 1
代码语言:javascript
运行
复制
sqldata = hiveSql.sql("""
        SELECT name,age from DB
""")

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

如代码所示,我的CSV如下所示:

代码语言:javascript
运行
复制
tom  16
jack 18

实际上,我需要CSV,如下所示

代码语言:javascript
运行
复制
name age
tom  16
jack 18
EN

回答 4

Stack Overflow用户

回答已采纳

发布于 2019-12-21 12:10:48

只需创建一个列名列表:

代码语言:javascript
运行
复制
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')

就是这样。完成了!

票数 1
EN

Stack Overflow用户

发布于 2019-12-21 12:17:29

还可以传入列名列表,如下所示

代码语言:javascript
运行
复制
tabel_df = pd.DataFrame(data=sqldata.collect(),columns=['name','age'])
票数 0
EN

Stack Overflow用户

发布于 2019-12-21 12:27:25

使用以下命令:

代码语言:javascript
运行
复制
tabel_df.to_csv('resultRDD.csv', header=['name','age'])
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/59433264

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档