python - 我怎样才能以有效的方式进行这个 numpy 数组操作？

Question

我是 python 新手。

如果我有一个（mxn）数组，我如何找到哪一列具有特定值的最大重复次数，例如。1 ? 是否有任何简单的操作，而不是编写迭代循环来做到这一点。

Answer 1

欢迎使用 python 和 numpy。您可以通过首先检查数组中的哪些值是 1，然后沿每一列计数，最后取 argmax 来获得具有最多 1 的列。在代码中，它看起来像这样：

>>> import numpy as np
>>> (m, n) = (4, 5)
>>> a = np.zeros((m, n))
>>> a[2, 3] = 1.
>>>
>>> a_eq_1 = a == 1
>>> repetitions = a_eq_1.sum(axis=0)
>>> np.argmax(repetitions)
3

或者更简洁：

>>> np.argmax((a == 1).sum(axis=0))
3