在 Pandas 数据框中计算唯一行的数量

Question

我需要计算pandas数据框中唯一行的数量。我尝试过这个解决方案，但它会产生错误。

这是我的代码：

import pandas as pd

df = {'x1': ['A','B','A','A','B','A','A','A'], 'x2': [1,3,2,2,3,1,2,3]}
df = pd.DataFrame(df)

print df.groupby(['x1','x2'], as_index=False).count()

这是错误：

Traceback (most recent call last):
  File "/home/user/workspace/project/test.py", line 9, in <module>
    print df.groupby(['x1','x2'], as_index=False).count()
  File "/usr/local/lib/python2.7/dist-packages/pandas/core/groupby.py", line 4372, in count
    return self._wrap_agged_blocks(data.items, list(blk))
  File "/usr/local/lib/python2.7/dist-packages/pandas/core/groupby.py", line 4274, in _wrap_agged_blocks
    index = np.arange(blocks[0].values.shape[1])
IndexError: list index out of range

我做错了什么？

- Danix Danilovic

Answer 1

使用size来完成（提示：您可以在末尾添加.reset_index()）

df.groupby(['x1','x2'], as_index=False).size()
Out[1262]: 
x1  x2
A   1     2
    2     3
    3     1
B   3     2
dtype: int64

或修复您的代码

df.groupby(['x1','x2'])['x2'].count()
Out[1264]: 
x1  x2
A   1     2
    2     3
    3     1
B   3     2
Name: x2, dtype: int64

如果您想了解唯一的分组，可以使用ngroups

df.groupby(['x1','x2']).ngroups
Out[1267]: 4

- BENY

---

谢谢，这很完美：print df.groupby(['x1','x2'], as_index=False).size().reset_index() - Danix Danilovic

在这种情况下，您可以使用更通用的代码：df.groupby(list(df.columns),as_index=False).size().reset_index()。 - Anton vBR

Answer 2

您可以删除重复项：

import pandas as pd

df = {'x1': ['A','B','A','A','B','A','A','A'], 'x2': [1,3,2,2,3,1,2,3]}
df = pd.DataFrame(df)

print(len(df.drop_duplicates()))

返回

4

- Anton vBR

Answer 3

如果要计算数据框中唯一行的出现次数，而不是使用count，现在应该使用value_counts。

df.groupby(['x1','x2'], as_index=False).value_counts()
Out[417]: 
  x1  x2  count
0  A   1      2
1  A   2      3
2  A   3      1
3  B   3      2

- user16836078