如何在groupby()之后将文件保存到csv文件.size()

时间:2016-01-31 22:19:23

标签: python csv size

groupby(['Id', 'event']).size()之后的数据:

1    A       3
     B       1
     C       6
2    A       3
     B       1

data.to_csv('data.csv', index=False)不会产生data.csv中我需要的格式,其中包含count列。

我在data.csv中需要的是

1    A       3
1    B       1
1    C       6
2    A       3
2    B       1

有什么想法吗?

1 个答案:

答案 0 :(得分:0)

df.to_csv(index=False)省略了输出中的索引。使用index=Trueheader=False(省略列名称):

import pandas as pd

df = pd.DataFrame({0: [1, 1, 1, 2, 2], 1: ['A', 'B', 'C', 'A', 'B'], 2: [3, 1, 6, 3, 1]})
df = df.set_index([0,1])

print(df.to_csv('data.csv', index=True, header=False, sep='\t'))

产量

1   A   3
1   B   1
1   C   6
2   A   3
2   B   1