Cách chuyển đổi mảng boolean thành mảng int

Tôi sử dụng Scilab và muốn chuyển đổi một mảng boolean thành một mảng các số nguyên:

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

Trong Scilab, tôi có thể sử dụng:

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

hoặc thậm chí chỉ cần nhân nó với 1:

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

Có một lệnh đơn giản cho việc này bằng Python hay tôi phải sử dụng một vòng lặp?

Mảng Numpy có một astypephương thức. Chỉ cần làm y.astype(int).

Lưu ý rằng thậm chí có thể không cần thiết để làm điều này, tùy thuộc vào những gì bạn đang sử dụng mảng. Bool sẽ được tự động khởi động thành int trong nhiều trường hợp, vì vậy bạn có thể thêm nó vào mảng int mà không cần phải chuyển đổi rõ ràng:

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

Các 1*yphương pháp làm việc trong NumPy quá:

>>> 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]) 

Nếu bạn đang yêu cầu một cách chuyển đổi danh sách Python từ Boolean sang int, bạn có thể sử dụng mapđể thực hiện:

>>> 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]

Hoặc sử dụng cách hiểu danh sách:

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

Sử dụng numpy, bạn có thể làm:

y = x.astype(int)

Nếu bạn đang sử dụng một mảng không phải numpy, bạn có thể sử dụng khả năng hiểu danh sách :

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

Hầu hết thời gian bạn không cần chuyển đổi:

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

Cách làm đúng là:

yourArray.astype(int)

hoặc là

yourArray.astype(float)

Tôi biết bạn đã yêu cầu các giải pháp không lặp lại, nhưng dù sao thì các giải pháp duy nhất mà tôi có thể đưa ra có thể là lặp nội bộ:

map(int,y)

hoặc là:

[i*1 for i in y]

hoặc là:

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

Một cách thú vị để làm điều này là

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