如何创建宽高为282、每个单元含256元素的NumPy三维数组？

嘿，我能理解你的困惑——NumPy数组的维度顺序有时候和我们直观理解的“宽”“高”对应关系容易搞混，尤其是当涉及到多通道数据的时候。

你想要的是一个宽282、高282的网格，每个网格单元包含256个元素的三维数组，本质上就是一个形状为(高, 宽, 通道数)或者(宽, 高, 通道数)的数组，具体可以根据你的使用场景调整：

• 如果你是在常规NumPy操作或大部分数据处理场景下使用：
  其实你最初写的np.zeros((282, 282, 256))已经完全满足需求了！这里前两个维度对应高和宽（数值都是282），第三个维度就是每个位置的256个元素。你觉得“未得到正确的宽和高”，大概率是因为你把数组的第一个维度当成了宽，但在空间数据（比如图像）的常规约定里，第一个维度代表高（行数），第二个维度代表宽（列数）。

• 如果你确实需要严格按照“宽在前、高在后”的顺序排列维度：
  因为你的宽高数值相同，直接用np.zeros((282, 282, 256))就没问题（顺序调换不影响结果）。要是以后遇到宽高不同的情况，你可以用np.transpose()来调整维度顺序，比如：

  # 示例：创建高282、宽300、每个单元256元素的数组
  arr = np.zeros((282, 300, 256))
  # 转换为宽300、高282、通道256的顺序
  arr_swapped = arr.transpose(1, 0, 2)
  

• 快速验证数组是否符合要求：
  你可以通过arr.shape查看维度，执行print(arr.shape)如果输出(282, 282, 256)，就说明前两个维度是你要的宽高，第三个是每个单元的元素数。还可以用print(len(arr[0][0]))验证每个单元的元素数量，输出应该是256。

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