如何按行对二维数组进行排序并将索引存储在新的二维数组中

Question

我有一个维度数组 (43, 25520)。

我想对这 43 列中的每一列进行排序，然后将每一行的排序索引存储在一个新的二维数组中。所以输出应该是一个新的排序 (43, 25520).

first_sorted = np.argsort(active_genevector[2,:])

这样我就得到了 43 行中第一行的排序后的 25520 个索引。

我怎样才能最终得到所有 43 个 (43, 25520) 数组？

Answer 1

我做了一个简短的脚本来测试你需要什么：

import numpy as np

test = np.random.random((43,25520))

result = np.argsort(test, axis = 1)

print(result.shape)

我认为这对你有用。