python - 以 pythonic 方式使用列表索引 numpy 数组

Question

假设我有一个像这样的 python 数组

array = [[2,3,4],[5,6,7],[8,9,10]]

还有一个列表

list = [0,2,1]

我基本上想要一个单行从列表给定的数组中提取索引元素

例如，对于给定的数组和列表:

result = [2,7,9]

我的下意识选择是

result = array[:, list]

但是没有用

我知道 for 循环应该可以，我只是想知道是否有一些索引可以解决这个问题

Answer 1

是这样的吗？

In [24]: a
Out[24]: 
array([[ 2,  3,  4],                                                                              
       [ 5,  6,  7],                                                                              
       [ 8,  9, 10]])                                                                             

In [25]: lis
Out[25]: [0, 2, 1]

In [26]: a[np.arange(len(a)), lis]                                                              
Out[26]: array([2, 7, 9])