np.array数组X=[0,1,0],Y=[[0,0],[1,1],[2,2],取出X中对应元素为0的索引,并将Y中对应这些索引的元素合并为数组Z=[[0,0],[2,2]],给出代码实现
时间: 2023-06-08 10:08:18 浏览: 87
多个数组元素集合到一个数组中并输出
以下是Python代码实现你的要求:
```python
import numpy as np
X = np.array([0, 1, 0])
Y = np.array([[0, 0], [1, 1], [2, 2]])
# 取出X中对应元素为0的索引
idx = np.where(X == 0)
# 将Y中对应这些索引的元素合并为数组Z
Z = Y[idx]
print(Z) # 输出 [[0, 0], [2, 2]]
```
解释一下代码逻辑:
首先,使用`np.where()`函数找出了`X`中对应元素为0的索引,并将其存储在`idx`变量中。
然后,将`Y`中对应这些索引的元素提取出来,形成一个新的数组`Z`。
最后,将`Z`输出以验证结果。
阅读全文