在Numpy中用2d数组索引2d数组
在Numpy中用2d数组索引2d数组
提问于 2020-04-16 19:55:21
我有个问题困扰了我几天。假设我们在Numpy中定义了一个2d数组：
x = np.array([[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])我们还定义了一个用于索引的一维数组，比方说：
ind = np.array([2,1])
如果我们尝试x[ind]，我们将得到：
array([[6, 7, 8],
[3, 4, 5]])这很有道理：x的第2行和第1行。
如果我们要运行：x[:,ind]，我们将得到：
array([[2, 1],
[5, 4],
[8, 7]])同样，这也很有意义--我们收到第2列，然后是第1列。
现在，我们将索引数组定义为2d：
ind = np.array([[2,1],
[2,2]])如果我们运行x[ind]，就会得到：
array([[[6, 7, 8],
[3, 4, 5]],
[[6, 7, 8],
[6, 7, 8]]])同样，这也是有意义的--对于索引2d数组中的每一行，我们收到一个2d数组，它表示原始2d数组x中对应的行。
但是，如果我们运行x[:,ind]，就会收到下一个数组：
array([[[2, 1],
[2, 2]],
[[5, 4],
[5, 5]],
[[8, 7],
[8, 8]]])我不理解这个输出，因为它返回索引行中的特定项，而不是整行。我假设，就像x[:,ind]在1d数组时的情况一样，我们将接收2d数组，其中包括原始x数组中的原始列。
回答 1
Stack Overflow用户
回答已采纳
发布于 2020-04-16 20:15:13
在最后一种情况下，使用索引数组：
print(ind)
array([[2, 1],
[2, 2]])因为ind是一个形状(2,2)的2D数组，并且您在第一个轴上取了一个完整的切片，使用ind，您将在每个行上按照A的列进行索引。例如，通过使用ind索引第二行ind，您将再次获得索引2->5、1->4、2->5和2->5中的元素，其结果形状与ind相同，因此是[[5,4][5,5]]。
它的每一行都是相同的，从而形成一个(3,2,2)形状的数组。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/61258958
复制相关文章
相似问题
腾讯云开发者
