首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >保存熊猫pivot_table以包含索引和列名

保存熊猫pivot_table以包含索引和列名
EN

Stack Overflow用户
提问于 2019-03-26 14:37:30
回答 1查看 5.6K关注 0票数 6

我想保存一个熊猫枢轴表供人类阅读,但DataFrame.to_csv不包括DataFrame.columns.name。我怎么能这么做?

示例

用于下列枢轴表:

代码语言:javascript
复制
>>> import pandas as pd
>>> df = pd.DataFrame([[1, 2, 3], [6, 7, 8]])
>>> df.columns = list("ABC")
>>> df.index = list("XY")
>>> df
   A  B  C
X  1  2  3
Y  6  7  8
>>> p = pd.pivot_table(data=df, index="A", columns="B", values="C")

查看透视表时,我们有索引名("A")和列名("B")。

代码语言:javascript
复制
>>> p
B    2    7
A
1  3.0  NaN
6  NaN  8.0

但是,当导出为csv时,我们会丢失列名:

代码语言:javascript
复制
>>> p.to_csv("temp.csv")

===temp.csv===
A,2,7
1,3.0,
6,,8.0

如何获得包含整个枢轴表(包括.columns.name ("B") )的某种人类可读的输出格式?

像这样的事就好了:

代码语言:javascript
复制
B,2,7
A,,
1,3.0,
6,,8.0
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2019-03-26 14:58:37

是的,通过追加助手DataFrame是可能的,但是读取文件有点复杂:

代码语言:javascript
复制
p1 = pd.DataFrame(columns=p.columns, index=[p.index.name]).append(p)
p1.to_csv('temp.csv',index_label=p.columns.name)
B,2,7
A,,
1,3.0,
6,,8.0

#set first column to index
df = pd.read_csv('temp.csv', index_col=0)
#set columns and index names
df.columns.name = df.index.name
df.index.name = df.index[0]
#remove first row of data
df = df.iloc[1:]
print (df)
B    2    7
A          
1  3.0  NaN
6  NaN  8.0
票数 7
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/55359807

复制
相关文章

相似问题

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