python array求大于4或小于0的索引
时间: 2024-03-27 20:39:24 浏览: 45
python 字符串操作(一)索引 切片
可以使用NumPy库来实现这个功能。以下是一个示例代码:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
indexes = np.where((arr > 4) | (arr < 0))[0]
print(indexes)
```
输出结果为:
```
[0 1 2 3 8 9]
```
其中,`np.where`函数返回一个元组,其中第一个元素是一个布尔数组,表示条件成立的位置,第二个元素是一个整数数组,表示条件成立的位置的索引。在这里,我们只需要使用第二个元素即可。`|`符号表示逻辑或。
阅读全文