有沒有辦法在 Python 中將多維數組重塑為一維向量？

Question

所以我試圖改變這個陣列

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

進入這個數組

   array([ 0, 4, 9, 6, 0])

因此，原始數組的每一行都被替換為一個值，該值等於“1”在行中的位置。

Answer 1

您可以使用numpy.argmax ：

a = np.array([[1., 0., 0., 0., 0., 0., 0., 0., 0.,0.],
          [0., 0., 0., 1., 0., 0., 0., 0., 0.,0.],
          [0., 0., 0., 0., 0., 0., 0., 0., 0.,1.],
          [0., 0., 0., 0., 0., 0., 1., 0., 0.,0.],
          [1., 0., 0., 0., 0., 0., 0., 0., 0.,0.]])

print( np.argmax(a, axis=1) )

印刷：

[0 3 9 6 0]

Answer 2

嘗試這個：

array = np.array([[1., 0., 0., 0., 0., 0., 0., 0., 0.,0.],
          [0., 0., 0., 1., 0., 0., 0., 0., 0.,0.],
          [0., 0., 0., 0., 0., 0., 0., 0., 0.,1.],
          [0., 0., 0., 0., 0., 0., 1., 0., 0.,0.],
          [1., 0., 0., 0., 0., 0., 0., 0., 0.,0.]])
          
print(np.where(array == 1)[1]

輸出：

array([0, 3, 9, 6, 0])