python - numpy 数组上的 2D argmax

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