如何基于变量创建指定维度的NumPy数组？

嘿，我懂你要做的事儿了——根据dim（维度数量）和terms（每个维度的长度）来动态创建numpy全零数组，之前尝试用字符串定义shape参数行不通对吧？那是因为numpy的np.zeros()函数需要的是整数构成的元组/列表作为shape参数，而不是字符串类型。

解决方法 #

核心思路很简单：我们只需要生成一个包含dim个terms的元组，把它传给np.zeros()就可以了。Python里可以用(terms,) * dim来快速生成这个元组——这里的逗号是关键，它确保我们创建的是单元素元组，再通过乘法扩展成指定长度的元组。

代码示例 #

import numpy as np

# 测试案例1：dim=2，terms=4
dim = 2
terms = 4
m = np.zeros((terms,) * dim)
print(m.shape)  # 输出: (4, 4)

# 测试案例2：dim=3，terms=4
dim = 3
terms = 4
m = np.zeros((terms,) * dim)
print(m.shape)  # 输出: (4, 4, 4)

# 测试案例3：dim=3，terms=3
dim = 3
terms = 3
m = np.zeros((terms,) * dim)
print(m.shape)  # 输出: (3, 3, 3)

为什么字符串方法不行？ #

如果你之前尝试把shape拼成字符串（比如"(4,4)"），那它本质是一个字符串对象，而np.zeros()需要的是能表示维度大小的整数序列（元组或列表）。直接传入字符串会触发类型错误，因为numpy无法把字符串解析成有效的维度参数。

内容的提问来源于stack exchange，提问作者Ryan O'Connor