40

我有一个 DataFrame，其索引city_id格式为 city [city],[state]（例如，new york,ny列中包含整数计数。问题是我有同一个城市的多行，我想city_id通过添加它们的列值来折叠共享 a 的行. 我看了看，groupby()但如何将它应用于这个问题并不是很明显。

编辑：

一个例子：我想改变这个：

city_id    val1 val2 val3
houston,tx    1    2    0
houston,tx    0    0    1
houston,tx    2    1    1

进入这个：

city_id    val1 val2 val3
houston,tx    3    3    2

如果有〜10-20k行。

4

2 回答 2

52

从...开始

>>> df
              val1  val2  val3
city_id                       
houston,tx       1     2     0
houston,tx       0     0     1
houston,tx       2     1     1
somewhere,ew     4     3     7

我可能会

>>> df.groupby(df.index).sum()
              val1  val2  val3
city_id                       
houston,tx       3     3     2
somewhere,ew     4     3     7

或者

>>> df.reset_index().groupby("city_id").sum()
              val1  val2  val3
city_id                       
houston,tx       3     3     2
somewhere,ew     4     3     7

第一种方法将索引值（在本例中为city_id值）传递给groupby并告诉它使用这些作为组键，第二种方法重置索引然后选择city_id列。有关更多示例，请参阅文档的这一部分。请注意，对象中还有许多其他方法DataFrameGroupBy：

>>> df.groupby(df.index)
<pandas.core.groupby.DataFrameGroupBy object at 0x1045a1790>
>>> df.groupby(df.index).max()
              val1  val2  val3
city_id                       
houston,tx       2     2     1
somewhere,ew     4     3     7
>>> df.groupby(df.index).mean()
              val1  val2      val3
city_id                           
houston,tx       1     1  0.666667
somewhere,ew     4     3  7.000000
于 2013-07-03T04:29:01.370 回答
3

在同一行中的东西。对不起，不是确切的复制品。

mydata = [{'subid' : 'B14-111', 'age': 75, 'fdg':1.78},
          {'subid' : 'B14-112', 'age': 22, 'fdg':1.56},{'subid' : 'B14-112', 'age': 40, 'fdg':2.00},]
df = pandas.DataFrame(mydata)

gg = df.groupby("subid",sort=True).sum()
于 2013-07-03T04:32:35.270 回答