python numpy convert bool to int

在 Python 的 NumPy 库中,我们可以使用 astype() 函数将布尔类型数组转换为整数类型数组。astype() 函数可以将给定数组转换为指定的数据类型。

例如,假设我们有以下的布尔类型的 NumPy 数组:

import numpy as np
arr = np.array([True, False, True, False])

我们可以使用 astype() 函数将该布尔类型的数组转换为整数类型数组,如下所示:

arr = arr.astype(int)

这将返回一个包含 1 和 0 的整数类型数组:

array([1, 0, 1, 0])

这样,我们就将布尔类型的数组成功转换为整数类型数组了。需要注意的是,astype() 函数会返回一个新的数组,而不会修改原始的数组。

  •