首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >如何在使用pandas和groupby时修复csv输出

如何在使用pandas和groupby时修复csv输出
EN

Stack Overflow用户
提问于 2019-05-10 03:36:41
回答 2查看 69关注 0票数 0

我有一个简单的csv,它有2列和大约50行。

第一列是ip,其他是cik,我想知道不同的cik有多少个ip。这是我的代码,它工作得很好:

代码:

代码语言:javascript
复制
import pandas as pd

csv = pd.read_csv('test.csv')
df = pd.DataFrame(csv)
df = df.groupby('cik').count()
df = pd.DataFrame(df).to_csv('output.csv', index=False)

但csv输出如下所示:

代码语言:javascript
复制
ip
49

我希望它类似于在groupby和count之后打印df值时的样子,如下所示:

因此,我在第一列中有cik,在其他列中有具有该cik的ip的数量。

EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2019-05-10 03:45:00

您的选项index=False使方法省略行名,在本例中为1515671,使用simple保存它:

代码语言:javascript
复制
df.to_csv('output.csv')
票数 2
EN

Stack Overflow用户

发布于 2019-05-10 03:43:14

尝试在输出to_csv之前添加reset_index。

代码语言:javascript
复制
import pandas as pd

csv = pd.read_csv('test.csv')
df = pd.DataFrame(csv)
df = df.groupby('cik').count().reset_index()   #reset_index creates 0...n index and avoids cik as index
df.to_csv('output.csv', index=False)

或者在输出to_csv时设置index=True

代码语言:javascript
复制
df.to_csv('output.csv', index=True)
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/56066451

复制
相关文章

相似问题

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