使用 shape=(dim,L,2) 访问 ndarray 的元素

Question

我目前正在使用 numpy.ndarray。我初始化了一个多维矩阵：

phi = np.zeros(dim_tuple(dim,L)+(2,))

其中 dim_tuple(dim,L) 返回具有昏暗时间 L 的元组 (L,L, ... ,L)。

我想创建一个作为参数的函数：

• 长度为 dim 的数组 x
• 一个数字 N：0 或 1

这将更新矩阵 phi 的元素 [x[0],x[1], ..., x[dim],N] 的值。

从现在开始，我尝试了这个：

index = np.append(x,N)

phi[index]=new_value

除了它不起作用。事实上，我检查它是否返回了这个命令：

phi[[x[0],x[1],...,x[dim],N]] = new_value

我试图在互联网上找到答案，但没有成功。

感谢您的帮助！

杰雷米

Answer 1

我想你想要的是：

phi[tuple(x) + (N,)] = new_value