假设我有一个numpy数组:
arr = np.array([1,2,3,4,4,5,3,2,10])
还有一个索引数组:
indices = np.array([0,1,4,6])
我可以编写一个简单的函数来完成这个任务,但我想知道是否有numpy内置函数像np.get(arr, indices)这样返回结果,例如在本例中,返回np.array([1,2,4,3])。
arr = np.array([1,2,3,4,4,5,3,2,10])
还有一个索引数组:
indices = np.array([0,1,4,6])
我可以编写一个简单的函数来完成这个任务,但我想知道是否有numpy内置函数像np.get(arr, indices)这样返回结果,例如在本例中,返回np.array([1,2,4,3])。
arr[indices]
是NumPy数组中基于索引进行切片的一种方式。它通过提供索引列表(可以是整数或布尔值)来获取数组中相应位置的元素。这些索引可以是单个数字、切片对象或一个包含数字和切片的元组。 - sascha