如何按一行对矩阵进行排序 - Python

Question

所以有一个这样的矩阵：[[0. 1. 0. 0. 0.][1. 0. 0. 0. 0.][0. 0. 0. 1. 0.][0. 0. 0. 0. 1.][1. 0. 3. 4. 4.]]如何按最后一行对其进行排序，以便列的内容保持不变，如下所示：[[1. 0. 0. 0. 0.][0. 1. 0. 0. 0.][0. 0. 0. 1. 0.][0. 0. 0. 0. 1.][0. 1. 3. 4. 4.]]现在我是这样做的：num_seq = 10seq_len = 5seq_width = 5con = np.random.randint(0, seq_width,size=seq_len)seq = np.zeros((seq_len, seq_width))seq[np.arange(seq_len), con] = 1seq[seq_len-1, np.arange(seq_width)] = conout = np.sort(seq, axis=1)并获得如下所示的输出：[[0. 0. 0. 0. 1.] [0. 0. 0. 0. 1.] [0. 0. 0. 0. 1.] [0. 0. 0. 0. 1.] [0. 1. 3. 4. 4.]]

Answer 1

你可以用np.argsort()一些 numpy 切片来做到这一点。使用您的示例数组：

arr = np.array([[0, 1, 0, 0, 0],

                [1, 0, 0, 0, 0],

                [0, 0, 0, 1, 0],

                [0, 0, 0, 0, 1],

                [1, 0, 3, 4, 4]])

arr[:, np.argsort(arr[-1, :])]

# array([[1, 0, 0, 0, 0],

#        [0, 1, 0, 0, 0],

#        [0, 0, 0, 1, 0],

#        [0, 0, 0, 0, 1],

#        [0, 1, 3, 4, 4]])

基本上按最后一行内容的升序值np.argsort(arr[-1, :])返回最后一行的索引。这将返回您给出的示例。arrarray([1, 0, 2, 3, 4])

然后我们只需使用arr[:, np.argsort(arr[-1, :])].

Answer 2

你可以用np.argsort()一些 numpy 切片来做到这一点。使用您的示例数组：

arr = np.array([[0, 1, 0, 0, 0],

                [1, 0, 0, 0, 0],

                [0, 0, 0, 1, 0],

                [0, 0, 0, 0, 1],

                [1, 0, 3, 4, 4]])

arr[:, np.argsort(arr[-1, :])]

# array([[1, 0, 0, 0, 0],

#        [0, 1, 0, 0, 0],

#        [0, 0, 0, 1, 0],

#        [0, 0, 0, 0, 1],

#        [0, 1, 3, 4, 4]])

基本上按最后一行内容的升序值np.argsort(arr[-1, :])返回最后一行的索引。这将返回您给出的示例。arrarray([1, 0, 2, 3, 4])

然后我们只需使用arr[:, np.argsort(arr[-1, :])].

Answer 3

使用 argsort 选择 con 作为索引：

import numpy as np

num_seq = 10

seq_len = 5

seq_width = 5

con = np.random.randint(0, seq_width,size=seq_len)

seq = np.zeros((seq_len, seq_width))

seq[np.arange(seq_len), con] = 1

seq[seq_len-1, np.arange(seq_width)] = con

i = np.argsort(con)

seq = seq[:,i]

print(con)

print(seq)