我有一个多维的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)
可能不同,我正在寻找一个相当通用的解决方案。希望我的问题能变得清晰!
1条答案
按热度按时间krcsximq1#
好吧,我觉得简单地用
应该可以了