Могу ли я расширить список списков в python?

Question

У меня есть данные, что выглядит следующим образом:

[[1, 100313], [2, 100313], [1, 100314], [3, 100315]] 

Со всем в формате строки (в противном случае я не могу итерацию)

Я хочу, чтобы запустить цикл, который смотрит через эти элементы и продолжается с [ 0, 0, 1], если он видит, второй элемент в списке, как 100313. Таким образом, в конечном счете, это будет выглядеть следующим образом:

[[1, 100313, 0, 0, 1], [2, 100313, 0, 0, 1], [1, 100314], [3, 100315]] 

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

for x, y in list: 
    if y == 100313: 
     list.extend([0,0,1]) 

и он не меняет этот список вообще. Почему это?

Answer 1

Вы можете сделать это с помощью list comprehension как в этом примере:

a = [[1, 100313], [2, 100313], [1, 100314], [3, 100315]] 

final = [k+[0,0,1] if k[1] == 100313 else k for k in a] 

print(final) 

Выход:

[[1, 100313, 0, 0, 1], [2, 100313, 0, 0, 1], [1, 100314], [3, 100315]] 

Обновление:

Вы попросили добавить еще один if ... else блок в одном list comprehension. Вы можете сделать это как в этом примере:

a = [[1, 100313], [2, 100313], [1, 100314], [3, 100315]] 
final = [k+[0,0,1] if k[1] == 100313 else k+[0,1,0] if k[1] == 100314 else k for k in a] 
print(final) 

Выход:

[[1, 100313, 0, 0, 1], [2, 100313, 0, 0, 1], [1, 100314, 0, 1, 0], [3, 100315]] 

Answer 2

Если вы имеете дело с большими объемами данных и хотите, чтобы избежать неэффективного использования памяти, что лучше использовать выражение генератора. Они выглядят очень похожи на список понимания для вместо квадратных скобок они имеют круглые скобки, за исключением:

from __future__ import print_function 

a = [[1, 100313], [2, 100313], [1, 100314], [3, 100315]] 
final = (k+[0,0,1] if k[1] == 100313 else k+[0,1,0] if k[1] == 100314 else k for k in a) 

for i in final: 
    print(i, end='') 

print() 

Выражения генератора возвращает объект генератора вместо целого списка. Вы можете лениво печатать элементы один за другим, перейдя через возвращаемый генератор.

Обратите внимание, что это имеет смысл только в том случае, если у вас действительно большие списки, а не как пример вашего вопроса.

Answer 3

Использовать список [x] [y] вместо оператора y in if. И используйте список [x], а не перечислите следующую строку. Это, безусловно, будет работать. Поскольку вы пытаетесь сделать это, вы пытаетесь расширить свой список, но то, что вы на самом деле хотите, - это расширение элемента списка. Надеюсь, это поможет.