用另一个 numpy 数组索引 numpy 数组 [英] Indexing numpy array with another numpy array

问题描述

假设我有

a = array([[1, 2],
           [3, 4]])

和

b = array([1,1])

我想在索引 a 中使用 b，即做 a[b] 并得到 4 而不是 [[3, 4], [3, 4]]

I'd like to use b in index a, that is to do a[b] and get 4 instead of [[3, 4], [3, 4]]

我大概可以做到

a[tuple(b)]

有更好的方法吗?

谢谢

推荐答案

根据 NumPy 教程，正确的做法是:

According the NumPy tutorial, the correct way to do it is:

a[tuple(b)]

这篇关于用另一个 numpy 数组索引 numpy 数组的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！