Установите нужный элемент на 1, остальные 0
Моя цель — установить только нужные элементы my_list в 1, а остальные — в 0.
Ожидаемый результат: [1,0,1]. Есть ли простой способ без использования явного цикла for?
my_list = [0.2, -0.5, 0.1]
set_to_one = [0, 2] # I want to set 0 th, 2nd element to 1, the rest 0
Ожидаемый результат [1,0,1].
Ответы 2
Возможно, создайте список нулей, а затем запустите цикл, установив нужные элементы в единицу.
my_list = [0] * len(my_list)
for i in set_to_one:
my_list[i] = 1
Результат:
>>> my_list
[1, 0, 1]
Вау, Рэймонд Хеттингер! После создания списка нулей, возможно, list(map(lambda x: my_list.__setitem__(x, 1), set_to_one)) сможет выполнить работу с циклом скрытыйfor :D
Используя starmap из itertools, вы можете распаковать кортежи, сгенерированные enumerate, а затем классифицировать.
import itertools as it
to_ones = [0, 2]
my_list = [0.2, -0.5, 0.1]
iter_ = it.starmap(lambda i, _:1 if i in to_ones else 0, enumerate(my_list))
print(list(iter_))
Примечание: поскольку выходной список не зависит от его элементов, а только от его длины, итератор, переданный в starmap, немного «избыточен». [Я повторно адаптировал свой первоначальный подход, в котором использовалось понимание списка]
Есть ли какая-то конкретная причина, по которой вы не хотите использовать простой цикл for?