Làm thế nào để chuyển đổi một mảng boolean để một mảng int

Question

tôi sử dụng Scilab, và muốn chuyển đổi một loạt các phép toán luận vào 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ỉ nhân nó bằng 1:

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

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

Answer 1

Mảng cố định có phương thức astype. Chỉ cần làm y.astype(int).

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

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

Answer 2

Phương pháp 1*y 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 là yêu cầu cho một cách để chuyển đổi danh sách Python từ Boolean để int, bạn có thể sử dụng map để làm điều đó:

>>> 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 danh sách comprehensions:

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

Answer 3

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 NumPy, bạn có thể sử dụng một list comprehension:

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

Answer 4

Tôi biết bạn yêu cầu không giải pháp -looping, nhưng các giải pháp duy nhất tôi có thể đưa ra có lẽ vòng lặp nội bộ anyway:

map(int,y) 

hay:

[i*1 for i in y] 

hay:

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

Answer 5

Phần lớn thời gian bạn không cần phải chuyển đổi:

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

Cách đúng để làm điều đó là:

yourArray.astype(int) 

hoặc

yourArray.astype(float)