python - 从 pandas 的数据框中提取唯一值和计数

Question

我需要提取唯一值、数据帧列中的计数以及数据帧中单独的索引键列。

让我们考虑以下示例，数据框 df1 组成

+------+--------+
| city |  score |
+------+--------+
| a    |     12 |
| b    |     45 |
| c    |     50 |
| d    |     65 |
| a    |     58 |
| a    |     70 |
+------+--------+

我需要在另一个数据帧 df2 中输出

+-------------+--------+--------------+
| unique city |  count |    index key |
+-------------+--------+--------------+
| a           |      3 |            1 |
| b           |      1 |            2 |
| c           |      1 |            3 |
| d           |      1 |            4 |
+-------------+--------+--------------+

我是 pandas 新手，我不知道如何做到这一点。

Answer 1

IIUC，您首先需要reset_index，以便我们可以将索引值包含在agg部分

df.reset_index().groupby('city').agg({'score':'count','index':'first'})
Out[404]: 
      index  score
city              
a         0      3
b         1      1
c         2      1
d         3      1