如何在NumPy三维矩阵中沿z轴打乱指定二维单元格？

给定形状为(c, b, a)的NumPy三维矩阵M，要打乱指定二维单元格(x, y)对应的z轴元素数组（即所有z层中(y, x)位置的元素组成的数组），完全不需要用展平、转置的繁琐方法，直接利用NumPy的索引特性就能简洁实现：

核心思路 #

NumPy中M[:, y, x]就是直接提取所有z轴（第一维）、y行（第二维）、x列（第三维）的元素，正好对应你需要的[M[z][y][x] for z in range(c)]数组，后续直接对这个数组做打乱操作即可。

两种实现方式 #

• 原地打乱（直接修改原矩阵）
  用np.random.shuffle()函数原地打乱目标数组，无需额外赋值：

  import numpy as np
  
  # 示例矩阵：形状(c=5, b=3, a=4)
  M = np.arange(5*3*4).reshape(5, 3, 4)
  x, y = 2, 1  # 指定要打乱的单元格位置
  
  # 提取目标z轴数组并原地打乱
  target_array = M[:, y, x]
  np.random.shuffle(target_array)
  # 此时M[:, y, x]已经是打乱后的结果
  

• 生成新数组赋值（保留原数组副本）
  如果不想原地修改原数组，用np.random.permutation()生成打乱后的新数组，再赋值回原矩阵位置：

  import numpy as np
  
  M = np.arange(5*3*4).reshape(5, 3, 4)
  x, y = 2, 1
  
  # 生成打乱后的数组并赋值
  M[:, y, x] = np.random.permutation(M[:, y, x])
  

优势说明 #

这种方法直接定位目标元素组，省去了展平、转置、重组矩阵的冗余步骤，代码更简洁，执行效率也更高，完全贴合你的需求。

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