如何高效创建首行全0末行全n的m列numpy数组？

安全阿岳

2026-7-29

高效生成特定结构NumPy数组的方法及命名问题

问题需求

需要生成一个m列的NumPy数组，满足以下结构：

数组的每一行是一个m维坐标，每个坐标分量的取值范围是0到n（包含n）
排列顺序为：第一列（最左）循环从0到n，完成一轮后第二列递增1，以此类推，最右列（第m列）变化最慢

当n=2、m=3时，目标数组如下：

[[0, 0, 0]
 [1, 0, 0]
 [2, 0, 0]
 [0, 1, 0]
 [1, 1, 0]
 [2, 1, 0]
 [0, 2, 0]
 [1, 2, 0]
 [2, 2, 0]
 [0, 0, 1]
 [1, 0, 1]
 [2, 0, 1]
 [0, 1, 1]
 [1, 1, 1]
 [2, 1, 1]
 [0, 2, 1]
 [1, 2, 1]
 [2, 2, 1]
 [0, 0, 2]
 [1, 0, 2]
 [2, 0, 2]
 [0, 1, 2]
 [1, 1, 2]
 [2, 1, 2]
 [0, 2, 2]
 [1, 2, 2]
 [2, 2, 2]]

要求：避免使用for循环，支持任意正整数n和m。

附加问题：这类数组是否有特定名称？

解决方案

可以通过NumPy的meshgrid或indices函数实现，完全无需循环：

方法1：使用`np.meshgrid`

利用meshgrid生成笛卡尔积网格，指定indexing='ij'保证坐标顺序，再将网格展平后按列堆叠：

import numpy as np

n = 2
m = 3

# 生成每个维度的取值数组（0到n）
axes = [np.arange(n + 1) for _ in range(m)]
# 生成笛卡尔积网格，indexing='ij'确保第一维变化最快
grid = np.meshgrid(*axes, indexing='ij')
# 将每个网格展平并按列拼接成最终数组
result = np.column_stack([g.ravel() for g in grid])

方法2：使用`np.indices`

通过indices生成网格索引，再按Fortran顺序展平（第一维变化最快），最后拼接成二维数组：

import numpy as np

n = 2
m = 3

# 生成m维网格的索引数组，每个维度大小为n+1
indices = np.indices((n + 1,) * m)
# 按Fortran顺序展平每个索引数组，再按列拼接
result = np.column_stack([idx.ravel(order='F') for idx in indices])

两种方法都能高效生成目标数组，时间复杂度为O((n+1)^m)，完全符合无循环的要求。