如何通过索引列表访问多维数组的元素

Question

我想通过定义的索引列表访问多维数组的特定元素（即张量）。假设我有一个 4 维数组“P”，索引列表“ind”定义为

 ind = [0,4,1,3]

它表征了它的元素之一。我想将“某物”分配给元素 P[0,4,1,3]。但是，当我这样做时

 P[ind] = "something"

Python 理解 P[[0,4,1,3]] 而不是 P[0,4,1,3]。

那么，我如何通过“ind”访问元素 P[0,4,1,3] ？

非常感谢您的帮助。

Answer 1

更新答案：

要通过给定的索引列表设置 N 维列表中的值，可以使用以下 setter 函数：

def set_val(ndim, ind, val):
    arr = ndim
    for i in ind[:-1]:
        arr = arr[i]
    arr[ind[-1]] = val

# usage:
P = [[[42]]]
set_val(P, [0, 0, 0,], 9001)

原答案：

要设置与普通 Python 列表的给定索引相对应的元素，您需要运行一个循环，如下所示：

arr = [1, 2, 3, 4, 5, 6, 7, 8]
ind = [0, 4, 1, 3]
val = 42
for i in ind:
    arr[i] = val

Plain Python 不提供对列表的向量操作。您可能会将 Python 与 Pandas 混淆。