Python中的布尔数组
时间: 2023-11-19 12:08:29 浏览: 60
Python中的布尔数组是由布尔值组成的一维数组,也称为布尔型数组。它的每个元素都是一个布尔值,可以是True或False。
在Python中,布尔数组可以使用NumPy库中的ndarray类型来创建。可以使用布尔运算符(如>,<,==等)来生成布尔数组。例如,以下代码创建了一个布尔数组:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
bool_arr = arr > 3
print(bool_arr)
```
输出:
```
[False False False True True]
```
在这个例子中,我们首先创建了一个NumPy数组arr,并使用比较运算符(>)来生成一个布尔数组bool_arr,该数组指示arr中的每个元素是否大于3。因为只有4和5大于3,所以bool_arr的最后两个元素为True,其余元素为False。
布尔数组可以用于过滤数组中的元素,例如:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
bool_arr = arr > 3
filtered_arr = arr[bool_arr]
print(filtered_arr)
```
输出:
```
[4 5]
```
在这个例子中,我们使用布尔数组bool_arr过滤了原始数组arr中的元素。只有bool_arr中为True的元素才会被保留,并存储在一个新的数组filtered_arr中。这种方法非常实用,可以方便地过滤数组中符合条件的元素。