Python中如何打乱数组首尾元素之外的其他元素？

问题背景 #

使用np.random.seed(0)生成固定均值和标准差的正态分布数组，尝试打乱数组中除首尾元素外的其他元素，但执行代码后中间元素位置未发生变化。

原代码 #

import numpy as np

np.random.seed(0)

mu, sigma = 50, 2.0 # 均值和标准差
Nodes=10
r = np.random.normal(mu, sigma, Nodes)

sort_r = np.sort(r); 
r1=sort_r[::-1]

r1=r1.reshape(1,Nodes)


r2 = r.copy()

np.random.shuffle(r2.ravel()[1:]) 

r2=r2.reshape(1,Nodes)                         # 实际半径值（单位：mu(m)）

maximum = r2.max()
indice1 = np.where(r2 == maximum)
r2[indice1] = r2[0][0]
r2[0][0] = maximum

r2[0][Nodes-1] = maximum  #+0.01*maximum

print("r2 with max at (0,0)=",[r2])

当前输出（多次运行结果一致） #

r2 with max at (0,0)= [array([[54.4817864 , 51.90017684, 53.52810469, 53.73511598, 48.04544424,
        51.95747597, 50.80031442, 50.821197  , 49.7935623 , 54.4817864 ]])]

期望输出（每次运行中间元素随机打乱） #

Run 1: r2 with max at (0,0)= [array([[54.4817864 , 53.52810469, 51.90017684, 53.73511598, 48.04544424,49.7935623 ,50.80031442, 50.821197  , 51.95747597,  54.4817864 ]])]

Run 2: r2 with max at (0,0)= [array([[54.4817864 , 51.90017684,53.52810469, 48.04544424, 53.73511598, 51.95747597,  49.7935623 ,50.80031442, 50.821197  , 54.4817864 ]])]

问题原因 #

np.random.shuffle(r2.ravel()[1:])这行代码中，r2.ravel()[1:]是原数组的切片副本，shuffle操作只会修改这个临时副本，不会影响原数组r2，因此原数组的中间元素位置没有变化。

修正后的代码 #

import numpy as np

np.random.seed(0)

mu, sigma = 50, 2.0 # 均值和标准差
Nodes=10
r = np.random.normal(mu, sigma, Nodes)

r2 = r.copy()

# 取出中间元素并打乱，直接修改原数组
middle = r2[1:-1]
np.random.shuffle(middle)
r2[1:-1] = middle

# 将最大值移到首尾
maximum = r2.max()
# 排除首元素，避免刚设置的最大值被替换
max_indices = np.where((r2 == maximum) & (np.arange(Nodes) != 0))[0]
r2[max_indices] = r2[0]
r2[0] = maximum
r2[-1] = maximum

# 调整为二维形状
r2 = r2.reshape(1, Nodes)

print("r2 with max at (0,0)=",[r2])

说明 #

1. 直接操作原数组的中间切片：通过r2[1:-1]获取中间元素的视图，打乱后直接赋值回原数组，确保修改生效。
2. 优化最大值替换逻辑：避免替换刚移到首位的最大值，保证首尾都能正确设置为最大值。
3. 若需要每次运行得到不同的打乱结果，移除np.random.seed(0)即可；保留该种子则每次打乱结果固定。

内容的提问来源于stack exchange，提问作者user20032724