在Python开头，将一个具有最大元素的Numpy阵列洗牌

Question

我正在尝试将数组r1洗牌，以便最大的元素始终位于r2 [0,0]。连接电流和可能的预期输出。

import numpy as np
r1 = np.array([[150.        , 132.5001244 , 115.00024881],
       [ 97.50037321,  80.00049761,  62.50062201],
       [ 45.00074642,  27.50087082,  10.00099522]])
r2=np.random.shuffle(r1)
print("r2 =",r2)

当前输出是

r2 = None

可能的预期输出

array([[150.,62.50062201,115.00024881],  
       [27.50087082,80.00049761,132.5001244]
       [10.00099522,97.50037321,45.00074642]])

Answer 1

为了清楚起见，您应该继续进行以下操作：

import numpy as np

r1 = np.array([[150.        , 132.5001244 , 115.00024881],
               [ 97.50037321,  80.00049761,  62.50062201],
               [ 45.00074642,  27.50087082,  10.00099522]])

# make a copy and shuffle the copy
r2 = r1.copy()
np.random.shuffle(r2.ravel())

# get the index of the max
idx = np.unravel_index(r2.argmax(), r2.shape)

# swap first and max
r2[idx], r2[(0, 0)] = r2[(0, 0)], r2[idx]

print(r2)

如果

最初排序阵列，则可以使用阵列的平坦视图：

r1 = np.array([[150.        , 132.5001244 , 115.00024881],
               [ 97.50037321,  80.00049761,  62.50062201],
               [ 45.00074642,  27.50087082,  10.00099522]])

r2 = r1.copy()

# r2.ravel() returns a view of the original array
# so we can shuffle only the items starting from 1
np.random.shuffle(r2.ravel()[1:])

可能的输出：

[[150.          80.00049761  62.50062201]
 [ 97.50037321 132.5001244  115.00024881]
 [ 45.00074642  27.50087082  10.00099522]]