不使用元组从numpy数组调用特定元素

8cdiaqws  于 2023-05-22  发布在  其他
关注(0)|答案(1)|浏览(163)

我有一个多维的numpy数组,类似于下面的一个:

dimensions = (3, 4, 5, 6) 
table = np.full(dimensions, np.nan)

使用tuple,从这个表中调用特定的元素并不是什么大问题,使用:element = tuple(1, 2, 3, 4) table[element]
但是,我尽量避免使用元组,因为我也在使用numba njit。有什么好的替代方案吗?
最理想的情况是,我会使用一个简单的numpy数组来调用特定的元素,即element = np.array([1, 2, 3, 4]) table[element](当然,它不能按预期工作。)如果表的尺寸和len(elements)可能不同,我正在寻找一个相当通用的解决方案。希望我的问题能变得清晰!

krcsximq

krcsximq1#

好吧,我觉得简单地用

element = np.array([1, 2, 3, 4], dtype=np.int64)
table[*element]

应该可以了

相关问题