Как векторизации столбец в числовом значении в Python

Question

Допустим, у меня есть столбец, содержащий три значения

a = ["Hello","Jello","Hello","Hello"] 

Я хочу, чтобы преобразовать его в

a = [1,0,1,1] 

Какой самый быстрый способ сделать это Python.

Answer 1

Я думаю, что понимание списка - это питонический способ справиться с этим.

a = [x == 'Hello' for x in a] 

Edit: Как уже упоминалось в комментариях, это будет возвращать список булевы, а не целые числа. Для целых чисел, необходимо изменить список понимание:

a = [1 if x == 'Hello' else 0 for x in a] 

или

a = [int(x == 'Hello') for x in a] 

Реквизит пойти в Марун Марун и Альберто Vassena для двух правильных решений.

Edit2: Я приуроченная обе версии, и вот результат:

Первый сценарий:

from timeit import default_timer as timer 

a = ['Hello', 'Jello', 'Hello', 'Hello'] 

start = timer() 
b = [1 if x == 'Hello' else 0 for x in a] 
end = timer() 
print b 
print('without cast: {}'.format(end - start)) 

Результат:

[1, 0, 1, 1] 
without cast: 2.86102294922e-06 

Второй сценарий:

from timeit import default_timer as timer 

a = ['Hello', 'Jello', 'Hello', 'Hello'] 

start = timer() 
b = [int(x == 'Hello') for x in a] 
end = timer() 
print b 
print('with cast: {}'.format(end - start)) 

Результат:

[1, 0, 1, 1] 
with cast: 5.96046447754e-06 

поэтому тест дает, что при использовании if else конструкция быстрее в этом случае.

Answer 2

я использовал бы список понимание таких как:

a = [(1 if 'Hello' == el else 0) for el in a] 

Но я думаю, что Себек, безусловно лучший. ;)

Answer 3

Это также основано на ответе Соббека - что, если мы не знаем элементов, присутствующих в массиве, перед рукой. Я думаю, это может быть решением:

a = ["Hello","Jello","Hello","Hello"] 

b = [int(x == list(set(a))[0]) for x in a ] 

b = [1, 0, 1, 1] 

Это был ответ, который я искал.