it-swarm.com.ru

Заменить значения в списке с помощью Python

У меня есть список, где я хочу заменить значения на None, где condition () возвращает True.

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

Например, если проверка условий bool (элемент% 2) должен вернуть:

[None, 1, None, 3, None, 5, None, 7, None, 9, None]

Какой самый эффективный способ сделать это?

108
ak.

Создайте новый список с пониманием списка:

new_items = [x if x % 2 else None for x in items]

Вы можете изменить исходный список на месте, если хотите, но на самом деле это не экономит время:

items = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
for index, item in enumerate(items):
    if not (item % 2):
        items[index] = None

Вот (Python 3.6.3) тайминги, демонстрирующие не временную экономию:

In [1]: %%timeit
   ...: items = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
   ...: for index, item in enumerate(items):
   ...:     if not (item % 2):
   ...:         items[index] = None
   ...:
1.06 µs ± 33.7 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

In [2]: %%timeit
   ...: items = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
   ...: new_items = [x if x % 2 else None for x in items]
   ...:
891 ns ± 13.6 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

И время Python 2.7.6:

In [1]: %%timeit
   ...: items = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
   ...: for index, item in enumerate(items):
   ...:     if not (item % 2):
   ...:         items[index] = None
   ...: 
1000000 loops, best of 3: 1.27 µs per loop
In [2]: %%timeit
   ...: items = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
   ...: new_items = [x if x % 2 else None for x in items]
   ...: 
1000000 loops, best of 3: 1.14 µs per loop
156
John Millikin
ls = [x if (condition) else None for x in ls]
61
Laurence Gonsalves

Перефразируя побочный вопрос, заданный ФП в комментарии, т.е.

что делать, если у меня был генератор, который дает значения из диапазона (11) вместо список. Можно ли заменить значения в генераторе?

Конечно, это тривиально просто ...

def replaceiniter(it, predicate, replacement=None):
  for item in it:
    if predicate(item): yield replacement
    else: yield item

Просто передайте любую итерацию (включая результат вызова генератора) в качестве первого аргумента, предикат, чтобы решить, должно ли значение быть заменено вторым аргументом, и дайте er rip.

Например:

>>> list(replaceiniter(xrange(11), lambda x: x%2))
[0, None, 2, None, 4, None, 6, None, 8, None, 10]
10
Alex Martelli

Вот еще один способ:

>>> L = range (11)
>>> map(lambda x: x if x%2 else None, L)
[None, 1, None, 3, None, 5, None, 7, None, 9, None]
9
balpha
>>> L = range (11)
>>> [ x if x%2 == 1 else None for x in L ]
[None, 1, None, 3, None, 5, None, 7, None, 9, None]
2
eduffy

Если вы хотите заменить значения на месте, вы можете Обновить исходный список со значениями из списка Понимание, назначив весь фрагмент оригинала.

data = [*range(11)] # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
id_before = id(data)
data[:] = [x if x % 2 else None for x in data]
data
# Out: [None, 1, None, 3, None, 5, None, 7, None, 9, None]
id_before == id(data)  # check if list is still the same
# Out: True

Если у вас есть несколько имен, указывающих на исходный список, например , Например, вы написали data2=data перед изменением списка И пропустили нотацию среза для назначения data, data будет перепривязана, чтобы указывать на вновь созданный список в то время как data2 все еще указывает на исходный неизмененный список.

data = [*range(11)] # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
data2 = data
id_before = id(data)
data = [x if x % 2 else None for x in data]  # no [:] here
data
# Out: [None, 1, None, 3, None, 5, None, 7, None, 9, None]
id_before == id(data)  # check if list is still the same
# Out: False
data2
# Out: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

Примечание. Это не рекомендация для общего предпочтения одного над другим .__ (изменение списка на месте или нет), но поведение, о котором вам следует знать.

0
Darkonaut