怎么把布尔数组返回为可以用的数组

如果你有一个布尔数组，你可以使用`np.where`函数将其转换为一个整数数组或浮点数数组。下面是一个使用`np.where`函数将布尔数组转换为整数数组的例子：

import numpy as np

# 假设有一个布尔数组 arr_bool
arr_bool = np.array([True, False, True, True])

# 使用 np.where 函数将 arr_bool 转换为整数数组
arr_int = np.where(arr_bool, 1, 0)

print(arr_int)

输出结果为：

[1 0 1 1]

在上面的代码中，`np.where(arr_bool, 1, 0)` 的意思是，将 arr_bool 中的 True 转换为 1，将 False 转换为 0，生成一个整数数组 arr_int。

下面是一个使用`np.where`函数将布尔数组转换为浮点数数组的例子：

import numpy as np

# 假设有一个布尔数组 arr_bool
arr_bool = np.array([True, False, True, True])

# 使用 np.where 函数将 arr_bool 转换为浮点数数组
arr_float = np.where(arr_bool, 1.0, 0.0)

print(arr_float)

输出结果为：

[1. 0. 1. 1.]

在上面的代码中，`np.where(arr_bool, 1.0, 0.0)` 的意思是，将 arr_bool 中的 True 转换为 1.0，将 False 转换为 0.0，生成一个浮点数数组 arr_float。