Python Pandas,从一个聚合多个列

时间:2014-11-10 17:26:11

标签: python pandas aggregate

我是熊猫新手,我有一个这样的DataFrame:

    name    value
0   alpha   a
1   beta    b
2   gamma   c
3   alpha   a
4   beta    b
5   beta    a
6   gamma   a
7   alpha   c

我想把它变成这种类型之一:

    name    a   b   c
0   alpha   2   0   1
1   beta    1   2   0
2   gamma   1   0   1

也就是说我希望按照#34; name"和"值",然后计算它们,并为"值"的每个值创建一列。我找到了。

1 个答案:

答案 0 :(得分:5)

这只是一个交叉表:

In [78]:

print pd.crosstab(df.name, df.value)
value  a  b  c
name          
alpha  2  0  1
beta   1  2  0
gamma  1  0  1

如果您使用groupby

In [90]:

print df.groupby(['name', 'value']).agg(len).unstack().fillna(0)
value  a  b  c
name          
alpha  2  0  1
beta   1  2  0
gamma  1  0  1

后者可能更快:

In [92]:

%timeit df.groupby(['name', 'value']).agg(len).unstack().fillna(0)
100 loops, best of 3: 3.26 ms per loop
In [93]:

%timeit pd.crosstab(df.name, df.value)
100 loops, best of 3: 7.5 ms per loop