python - Pandas ，验证数据，检查所有组是否具有相同的长度

Question

具有以下 DF:

   A  B
0  1  2
1  1  2
2  4  3
3  4  3
4  5  6
5  5  6
6  5  6

与列分组后 A我得到 3 组

(1,    A  B
0  1  2
1  1  2)
(4,    A  B
2  4  3
3  4  3)
(5,    A  B
4  5  6
5  5  6
6  5  6)

我想计算与特定行数不同的组，例如 2 作为输入将导致 1 作为输出，因为只有 1 个组有 3 行，而 3 作为输入将为其他组输出 2。

   A  B
0  1  2
1  1  2

2  4  3
3  4  3

4  5  6
5  5  6
6  5  6

对于这样的任务，Pandas 的解决方案是什么？

Answer 1

我想你需要 Series.value_counts 测试不等于 Series.ne 然后计算 True 的数量来自 sum :

N = 2
a = df['A'].value_counts().ne(N).sum()
print (a)
1