我有两个numpy数组,X=[1 2 3 4 5 6 7 8 9],Y=[0 2 5],现在我想要获得X数组中Y下标的数据,即得到一个numpy数组为Z=[1 3 6],请问如何实现
时间: 2024-05-08 19:15:41 浏览: 46
可以使用numpy的take函数来实现:
```python
import numpy as np
X = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
Y = np.array([0, 2, 5])
Z = np.take(X, Y)
print(Z)
```
输出结果为:
```python
[1 3 6]
```
take函数的第一个参数是要取值的数组,第二个参数是要取值的下标数组。
阅读全文