numpy count元素跨轴0匹配来自另一个数组的值

Question

给出一个3D数组，如：

auto_delete=true

和0轴上的最大值数组：

array = np.random.randint(1, 6, (3, 3, 3))

是否有矢量化方式来计算数组0轴中元素的数量，这些元素的数量等于max_array中匹配索引的值？例如，如果数组在一个轴0位置包含[1,3,3]，则输出为2，对其他8个位置依此类推，返回带有计数的数组。

Answer 1

要计算x中等于xmax中相应值的值的数量，您可以使用：

(x == xmax).sum(axis=0)

请注意，由于x具有形状（3,3,3）且xmax具有形状（3,3），因此表达式x == xmax会导致NumPy变为broadcast { {1}}最多为形状（3,3,3），左侧添加了新轴。

例如，

xmax