Я хотел бы удалить несколько элементов в списке по их индексу:
dS = [0, 0.02, 0, 0.04, 0.07, 0]
dN = [1, 0.02, 0.3, 0.7, 0.9]
pos = [i for i, e in enumerate(dS) if e ==0]
И теперь я хотел бы удалить (в обоих списках) все элементы на позициях 0 в первом списке.
pos = (i for i, e in enumerate(dS) if e ==0) и после, пробежался по списку и удалил каждое: for idx in pos[::-1]: del dS[idx] Видите ли, я перевернул список, чтобы не менять числа, которые будут удален 30.11.2018 [dN[i] for i, e in enumerate(dS) if e] 30.11.2018 Предполагая, что у вас есть old_list со списком индексов pos, от которых вы хотите избавиться:
new_list = [old_list[i] for i, e in enumerate(old_list) if i not in pos]
Это будет работать как для списка в вашем вопросе, избавившись от элемента в индексе, указанном pos, просто замените old_list именем списка, которое у вас есть в настоящее время:
dS = [0, 0.02, 0, 0.04, 0.07, 0]
dN = [1, 0.02, 0.3, 0.7, 0.9]
pos = [i for i, e in enumerate(dS) if e ==0]
dS = [dS[i] for i, e in enumerate(dS) if i not in pos]
dN = [dN[i] for i, e in enumerate(dN) if i not in pos]
>>> dS, dN
([0.02, 0.04, 0.07], [0.02, 0.7, 0.9])
Это отлично работает для списков разной длины, как показано в вашем случае выше.
Я хотел бы удалить (в обоих списках) все элементы на позициях 0 в первом списке.
Хорошо, сначала обработайте второй список, выбрав только элементы того же индекса, что и ненулевые элементы первого списка:
dN = [dN[i] for i, e in enumerate(dS) if e]
затем сделайте то же самое для dS (примечание: в этом случае вариант будет использовать filter):
dS = list(filter(None,dS))
dN больше, это решение урежет его элементы в конечном результате. 30.11.2018 У вас есть выбор: создать новые копии каждого списка, опустив некоторые элементы, или изменить оба списка на месте.
dS = [0, 0.02, 0, 0.04, 0.07, 0]
dN = [1, 0.02, 0.3, 0.7, 0.9]
# make copies of both lists while omitting indices
# that corresponds to zeros in dS
dS, dN = map(list, zip(*[(s, n) for s, n in zip(dS, dN) if s != 0]))
# alternatively, modify lists inplace
dS = [0, 0.02, 0, 0.04, 0.07, 0]
dN = [1, 0.02, 0.3, 0.7, 0.9]
# process lists in reverse
for i in range(1, len(dS)):
try:
if dS[-i] == 0:
del dS[-i], dN[-i]
except IndexError:
break
Поскольку ваши списки имеют разную длину, вам нужно только пройти через свой список pos и удалить из обоих списков, например: (важно запускать в обратном порядке в pos, чтобы гарантировать, что вы не попытаетесь удалить индекс элемента, который не уже существует)
for idx in pos[::-1]:
del dS[idx]
if idx < len(dN):
del dN[idx]
dSиdNне зависит друг от друга. 30.11.2018pos, вероятно, должен бытьsetдля более быстрого поиска 01.12.2018dS = [e for i, e in enumerate(dS) if i not in pos]01.12.2018