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数组，则可以使用list comprehension：

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

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

map(int,y)

或：

[i*1 for i in y]

或：

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

Answer 6

一个有趣的方法是

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

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

6 个答案: