Question

从numpy数组v开始：

v = \
np.array([[0, 0, 0, 0, 0],
          [0, 0, 1, 0, 0],
          [1, 0, 0, 0, 0],
          [0, 1, 0, 0, 0],
          [0, 0, 1, 0, 1],
          [0, 0, 0, 0, 1],
          [0, 0, 0, 1, 1]])

我想在每列中找到第一个非零值，并将这些值乘以100。

我想要的结果是：

array([[  0,   0,   0,   0,   0],
       [  0,   0, 100,   0,   0],
       [100,   0,   0,   0,   0],
       [  0, 100,   0,   0,   0],
       [  0,   0,   1,   0, 100],
       [  0,   0,   0,   0,   1],
       [  0,   0,   0, 100,   1]])

我想通过沿每个轴取argmax来解决这个问题：

i = v.argmax(0)
j = v.argmax(1)
v[i, j] *= 100

我知道我没有正确使用i和j，所以我该如何解决这个问题？

Answer 1

您只想浏览列

v[v.argmax(axis=0), np.arange(v.shape[1])] *= 100

numpy数组上的2D argmax

1 个答案: