python – 作为多维数组索引的元组

前端之家收集整理的这篇文章主要介绍了python – 作为多维数组索引的元组前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我找到了一个与我非常相似的问题,但不完全相同.
这一个: here
然而,在ntimes的情况下,数组的大小与元组指向的维度的数量相匹配.
在我的情况下,我有一个4维数组和一个二维元组,就像这样:
from numpy.random import rand
big_array=rand(3,3,4,5)
tup=(2,2)

我想使用元组作为前两个维度的索引,并手动索引最后两个维度.就像是:

big_array[tup,2]

但是,我沿着第四维获得了索引= 2的第一维的重复(因为它在技术上没有被索引).这是因为这个索引将双索引解释为第一维而不是每个维的一个值,

eg. 
| dim 0:(index 2 AND index 2),dim 1:(index 3),dim 2:(index 2),dim 3:(no index)|
instead of 
|dim 0(index 2),dim 1(index 2),dim 2:(index 3),dim 3:(index 2)|.

那我怎么能“打开”这个元组呢?有任何想法吗?
谢谢!

解决方法

您也可以单独传递您的第一个元组以获得感兴趣的切片,然后将其分别编入索引:
from numpy.random import rand
big_array=rand(3,5)
chosen_slice = (2,2)

>>> big_array[ chosen_slice ]
array([[ 0.96281602,0.38296561,0.59362615,0.74032818,0.88169483],[ 0.54893771,0.33640089,0.53352849,0.75534718,0.38815883],[ 0.85247424,0.9441886,0.74682007,0.87371017,0.68644639],[ 0.52858188,0.74717948,0.76120181,0.08314177,0.99557654]])

>>> chosen_part = (1,1)

>>> big_array[ chosen_slice ][ chosen_part ]
0.33640088565877657

对于某些用户而言,这可能稍微更具可读性,但除此之外,我倾向于mgilson的解决方案.

原文链接:https://www.f2er.com/python/186629.html

猜你在找的Python相关文章