python - 如何使用 numpy 对数组和矩阵进行排序

Question

我有一个 CSR 稀疏的数组y和矩阵X。

我需要做一个随机排序y和 X ，其中每一行y对应于X.

使用 NumPy，我该怎么做？

Answer 1

为了获得随机排序，您可以尝试使用该np.random.shuffle功能。

# Create an array of indices
indices = np.arange(y.size)
# Randomize it
np.random.shuffle(indices)

您现在可以使用花哨的索引indices来随机化。yy_new = y[indices]

您可以使用相同indices的方法对矩阵进行重新排序，但请注意，CSR 矩阵不支持花哨的索引。您必须将其转换为 LIL，对其重新排序，然后将其重新转换为 CSR。