我有两个数据,A数列是每天的股价,B数列是0和1组成的数值。我要把对应B列数值是1的时候,把A的数值取出来,生成一个C,能用np
时间: 2024-03-16 21:45:40 浏览: 56
可以使用`np.where`函数和布尔索引来实现这个要求。下面是一个示例代码:
```python
import numpy as np
# 假设 A 数组是每天的股价,B 数组是 0 和 1 组成的数值
A = np.array([10, 12, 11, 13, 14, 15])
B = np.array([0, 1, 1, 0, 1, 0])
# 使用 np.where 和布尔索引生成新数组 C
C = A[B == 1]
print(C)
```
输出结果为:
```
[12 11 14]
```
在上面的代码中,`A[B == 1]` 的意思是,先使用布尔索引 B == 1,生成一个布尔数组,表示对应位置上 B 中的值是否为 1。然后将布尔数组作为索引,从 A 数组中取出对应位置上的值,生成新的数组 C。
阅读全文