如何使用numpy将数组和矩阵排序在一起

Question

我有一个数组y和CSR稀疏矩阵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随机化y花哨的索引y_new = y[indices]。

您可以使用相同的indices重新排列矩阵，但要小心，CSR矩阵不支持花哨的索引。您必须将其转换为LIL，对其重新排序，然后将其重新转换为CSR。