python - 如何将布尔数组转换为int数组

Question

我使用 Scilab，并希望将布尔数组转换为整数数组：

>>> x = np.array([4, 3, 2, 1])
>>> y = 2 >= x
>>> y
array([False, False,  True,  True], dtype=bool)

在 Scilab 我可以使用：

>>> bool2s(y)
0.    0.    1.    1.  

甚至只是将其乘以 1：

>>> 1*y
0.    0.    1.    1.  

在 Python 中是否有一个简单的命令，或者我必须使用循环？

Answer 1

Numpy 数组有一个astype方法。做吧y.astype(int)。

请注意，甚至可能没有必要这样做，具体取决于您使用数组的目的。在许多情况下，Bool 会自动提升为 int，因此您可以将其添加到 int 数组中，而无需显式转换：

>>> x
array([ True, False,  True], dtype=bool)
>>> x + [1, 2, 3]
array([2, 2, 4])

Answer 2

该1*y方法也适用于 Numpy：

>>> import numpy as np
>>> x = np.array([4, 3, 2, 1])
>>> y = 2 >= x
>>> y
array([False, False,  True,  True], dtype=bool)
>>> 1*y                      # Method 1
array([0, 0, 1, 1])
>>> y.astype(int)            # Method 2
array([0, 0, 1, 1]) 

如果您正在寻求一种将 Python 列表从 Boolean 转换为 int 的方法，您可以使用map它来执行此操作：

>>> testList = [False, False,  True,  True]
>>> map(lambda x: 1 if x else 0, testList)
[0, 0, 1, 1]
>>> map(int, testList)
[0, 0, 1, 1]

或使用列表推导：

>>> testList
[False, False, True, True]
>>> [int(elem) for elem in testList]
[0, 0, 1, 1]

Answer 3

使用 numpy，您可以执行以下操作：

y = x.astype(int)

如果您使用的是非 numpy 数组，则可以使用列表推导：

y = [int(val) for val in x]

Answer 4

大多数时候你不需要转换：

>>>array([True,True,False,False]) + array([1,2,3,4])
array([2, 3, 3, 4])

正确的做法是：

yourArray.astype(int)

或者

yourArray.astype(float)

Answer 5

一个有趣的方法是

>>> np.array([True, False, False]) + 0 
np.array([1, 0, 0])

Answer 6

我知道你要求非循环解决方案，但我能想出的唯一解决方案可能是内部循环：

map(int,y)

或者：

[i*1 for i in y]

或者：

import numpy
y=numpy.array(y)
y*1