获取numpy数组的第n个和第m个元素[重复]

【问题标题】：Get nth and mth elements of a numpy array [duplicate]获取numpy数组的第n个和第m个元素[重复]
【发布时间】：2020-09-03 18:43:25
【问题描述】：

一个非常基本的问题，但我在这里通过谷歌搜索找不到类似的问题。

tmp = np.array([1,2,3,4,5])

我可以通过tmp[1] 提取 2 和通过tmp[1:4] 提取 2 到 4

假设我想提取 2 AND 4。最简单的方法是什么？

【问题讨论】：

【解决方案1】：

您可以使用.take()

import numpy as np
tmp = np.array([1,2,3,4,5]).take([1,4])
# Out[4]: (2, 5)

【讨论】：