使用索引数组python从张量中获取值

Question

我有一个数组 A ：

A = [[1,  2  ,3  ,4],
     [5,  6  ,7  ,8],
     [9, 10 ,11 ,12],]

我想获得第三个元素中的第二行（即'7'）：

我可以这样做：

A[1,2]

对于一般尺寸编号，我想要一些通用的东西。 给定索引列表 B=[1,2] 我想要像 MATLAB 索引这样的东西：

A[B] 或 A[*B]

第一个给出 2 行，第二个导致错误。 我怎样才能做到这一点？

---

编辑：类型（A）=类型（B）=np.array

Answer 1

您不需要使用“*”解包集合，因为 numpy 支持元组索引。您的问题是按列表索引和按元组索引之间存在差异。所以，

import numpy as np

A = np.array([
  [1, 2, 3, 4],
  [5, 6, 7, 8],
  [9, 10, 11, 12]
])

print("indexing by list:
", A[[0, 1]], '
')
# Output:
# indexing by list:
#   [[1 2 3 4]
#   [5 6 7 8]] 

print("indexing by tuple:
", A[(0, 1)])
# Output:
# indexing by tuple:
#   2

希望能帮助到你。 For the further reading 关于 numpy 索引。

Answer 2

作为替代方案，假设您对“array indexing”感兴趣，即获取长度为L 的值数组，其中数组中的每个值都来自A 中的位置。您想通过某种索引来指示这些位置。

假设A 是N 维。数组索引通过让您传递长度为L 的N 列表的元组来解决这个问题。 ith 列表给出了与A 中的ith 维度相对应的索引。在最简单的情况下，索引列表的长度可以是一（L=1）：

>>> A[[1], [2]]
array([7])

但索引列表可能更长（L=5）：

>>> A[[1,1,1,1,0], [2,2,2,2,0]]
array([7, 7, 7, 7, 1])

我强调这是一个列表元组；你也可以去：

>>> first_axis, second_axis = [1,1,1,1,0], [2,2,2,2,0]
>>> A[(first_axis,second_axis)]
array([7, 7, 7, 7, 1])

您还可以传递非元组序列（列表列表而不是列表元组）A[[first_axis,second_axis]]，但在 NumPy 版本 1.21.2 中，给出了 FutureWarning，并指出已弃用。

另一个例子：https://stackoverflow.com/a/23435869/4901005