计算每行中等于每行中第一个元素的元素数

Question

a = np.array((['1', '1', '1', '2'],
              ['3', '3', '3', '3'], 
              ['2', '1', '1', '2'], 
              ['1', '3', '1', '2']))

我正在寻找一种方法（而不是遍历数组）来计算每行中等于该行第一个元素的元素数。 即在第一行中，第一个元素是'1' ，并且该值还有2次出现。 最后的结果应该是：

result = [2, 3, 1, 1]

我的数组可能非常大，因此遍历每一行会很慢。

Answer 1

您可以使用列表理解来做到这一点：

result = [list(a[i,:]).count(a[i,0])-1 for i in range(a.shape[0])]
print(result)

这给了我：

[2, 3, 1, 1]

Answer 2

与切片a[..., 0]进行比较。 Append 一个新轴，使其再次具有正确的尺寸。

result = (a == a[..., 0][..., np.newaxis]).sum(axis=-1) - 1

...确保这适用于任何维度的 arrays，而使用:仅适用于 2D。