如何使用基于变量的维度创建数组？

Question

我有两个变量：dim和terms。我想使用这些变量来定义数组。例如，

• 如果dim=2和terms=4，我想m=np.zeros((4,4))
• 如果dim=3和terms=4，我想m=np.zeros((4,4,4))
• 如果dim=3和terms=3，我想m=np.zeros((3,3,3))

我尝试使用字符串定义参数，但它并不接受。

Answer 1

您可以使用：

np.zeros((terms,) * dim)