it-swarm.com.ru

Как преобразовать логический массив в массив int

Я использую 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, или я должен был бы использовать цикл?

92
Kwolf

Массивы Numpy имеют метод astype. Просто сделайте y.astype(int).

Обратите внимание, что это может даже не понадобиться, в зависимости от того, для чего вы используете массив. Во многих случаях Bool автоматически переводится в int, поэтому вы можете добавить его в массивы int без явного преобразования:

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

Метод 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 из логического значения в 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]
39
Sukrit Kalra

Используя numpy, вы можете сделать:

y = x.astype(int)

Если бы вы использовали непустой массив, вы могли бы использовать список понимания :

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

Большую часть времени вам не нужно преобразование:

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

Правильный способ сделать это:

yourArray.astype(int)

или же

yourArray.astype(float)
10
Gioelelm

Я знаю, что вы спрашивали о решениях, не использующих циклы, но единственные решения, которые я могу придумать, вероятно, в любом случае внутри цикла:

map(int,y)

или же:

[i*1 for i in y]

или же:

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