从numpy数组中剪切特定元素的Pythonic方法

Question

我有一个Python列表(numpy数组)和另一个列表，它包含了我想保留的第一个数组的值位置的索引。

有什么毕达通的方法吗？我知道numpy.delete，但我想保留这些元素，而不是删除它们。

Answer 1

最简单的方式可能也是最简单的：

a = np.array([2,5,6,3,6,3,45,6])
b = [0,3,4,7] # indices that you need to keep
c = a[b]

或者，如果不再需要a：

a = a[b]

Answer 2

您可以使用要保留的值创建一个新列表。

a = np.array([2,5,6,3,6,3,45,6])
b = [0,3,4,7] #indices that yo need to keep
c = [a[i] for i in b]

Answer 3

为什么不使用c=a[b]，因为这是从数组a获取值的方式。