编程语言
首页 > 编程语言> > 有问题将python数据帧汇总到每条记录一行

有问题将python数据帧汇总到每条记录一行

作者:互联网

我在表单中有一个数据框:

df = pd.DataFrame({'id':['a', 'a', 'a', 'b','b'],'var':[1,2,3,5,9]})

我试图重塑它,以便每个’id’有一行,值’var’显示在一行中,所以’a’会有1,2,3 …’b’会有’5,9′

我尝试过:

test = pd.crosstab(df.id, df.var)  # but it does not work?

如果有人可以帮助我,我将不胜感激

编辑,我把所需的结果作为图片包含在这里enter image description here

解决方法:

更新:

In [32]: df.groupby('id')['var'].apply(lambda x: x.astype(str).str.cat(sep=',')).reset_index()
Out[32]:
  id    var
0  a  1,2,3
1  b    5,9

或将var作为列表:

In [29]: df.groupby('id')['var'].apply(list).reset_index()
Out[29]:
  id        var
0  a  [1, 2, 3]
1  b     [5, 9]

老答案:

您可以在内部使用crosstab()方法使用的pivot_table()IIUC吗?

In [26]: df.pivot_table(index='id', columns='var', aggfunc='size', fill_value=0)
Out[26]:
var  1  2  3  5  9
id
a    1  1  1  0  0
b    0  0  0  1  1

标签:python,pandas,dataframe,pivot-table
来源: https://codeday.me/bug/20190828/1756318.html