Python "not in" index out of range enrror

Question

Я имею дело с какой-то проблемой, и я не смог найти решение. Моя проблема заключается в том, что я контролирую значение во вложенном списке, если он не находится в другом списке и удаляет его, если его нет, но в не в строке он дает мне ошибку, такую ​​как индекс за пределами диапазона.

def heroes_updater(men_pref,women): 
    for i in range(0,len(men_pref)-1): 
     for j in range(0,len(men_pref[i])-1): 
      if men_pref[i][j] not in women: 
       men_pref[i]=men_pref[i][:j]+men_pref[i][j+1:] 

пример men_pref:

[['Storm', 'Black Widow', 'Scarlet Witch', 'Rouge', 'Mystique', 'Jean Grey', 'Ms. Marvel', 'Gamora', 'Invisible Woman', 'Elektra'], ['Storm', 'Elektra', 'Jean Grey', 'Scarlet Witch', 'Mystique', 'Ms. Marvel', 'Gamora', 'Rouge', 'Black Widow', 'Invisible Woman'], ['Invisible Woman', 'Scarlet Witch', 'Mystique', 'Black Widow', 'Ms. Marvel', 'Elektra', 'Jean Grey', 'Gamora', 'Storm', 'Rouge']] 

пример женщины:

['Jean Grey', 'Elektra', 'Mystique', 'Ms. Marvel', 'Rouge'] 

И ошибка:

if men_pref[i][j] not in women: 

IndexError: list index out of range

Answer 1

Удаляя элементы из вашего списка, список становится короче. j больше, чем длина списка. Чтобы обойти эту проблему, просто не изменять списки, но создать новый:

def heroes_updater(men_pref,women): 
    result = [] 
    for prefs in men_pref: 
     new_prefs = [] 
     for pref in prefs: 
      if pref in women: 
       new_prefs.append(pref) 
     result.append(new_prefs) 
    men_pref[:] = result 

или лучше:

def filter_non_heroes(men_pref,women): 
    return [ 
     [pref for pref in prefs if pref in women] 
     for prefs in men_pref 
    ] 

Answer 2

Вы редактировать в списке, который вы читаете, вы никогда не должны этого делать.

С линией men_pref[i]=men_pref[i][:j]+men_pref[i][j+1:]
вы удалением элемента из списка men_pref[i], но ваша j переменный выходит из 0 к первоначальной длине списка, так что вы будете в конечном счете есть ошибка индекса при регистрации для men_pref[i][j] если j>len(men_pref[i])

EDIT: из, если вы хотите изменить ваш текущий список, то вы должны будете читать его с обратной индекса (вы начинаете с последнего, если это не входит в список женщин, удалить а затем продолжить со следующего пункта):

def heroes_updater(men_pref,women): 
    for i in range(len(men_pref)-1, -1,-1): 
     for j in range(len(men_pref[i])-1, -1, -1): 
      if men_pref[i][j] not in women: 
       men_pref[i].pop(j) 
       # An alternative: del(mem_pref[i][j]) 

Другим способом было бы использовать список понимание:

def heroes_updater(men_pref,women): 
    for i in range(len(men_pref)-1, -1,-1): 
     mem_pref[i] = [_w_ for _w_ in mem_pref[i] if _w_ in women] 

Есть и другие варианты, но я оставлю это для вас. Вот как вы учитесь.

Answer 3

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

men = [['Storm', 'Black Widow', 'Scarlet Witch', 'Rouge', 'Mystique', 'Jean Grey', 'Ms. Marvel', 'Gamora', 'Invisible Woman', 'Elektra'], ['Storm', 'Elektra', 'Jean Grey', 'Scarlet Witch', 'Mystique', 'Ms. Marvel', 'Gamora', 'Rouge', 'Black Widow', 'Invisible Woman'], ['Invisible Woman', 'Scarlet Witch', 'Mystique', 'Black Widow', 'Ms. Marvel', 'Elektra', 'Jean Grey', 'Gamora', 'Storm', 'Rouge']] 

wom = {'Jean Grey', 'Elektra', 'Mystique', 'Ms. Marvel', 'Rouge'} 

men[:] = [[ele for ele in sub if ele in wom] for sub in men] 

print(men) 

Или функционально, если порядок не имеет значения:

men = [['Storm', 'Black Widow', 'Scarlet Witch', 'Rouge', 'Mystique', 'Jean Grey', 'Ms. Marvel', 'Gamora', 'Invisible Woman', 'Elektra'], ['Storm', 'Elektra', 'Jean Grey', 'Scarlet Witch', 'Mystique', 'Ms. Marvel', 'Gamora', 'Rouge', 'Black Widow', 'Invisible Woman'], ['Invisible Woman', 'Scarlet Witch', 'Mystique', 'Black Widow', 'Ms. Marvel', 'Elektra', 'Jean Grey', 'Gamora', 'Storm', 'Rouge']] 

wom = {'Jean Grey', 'Elektra', 'Mystique', 'Ms. Marvel', 'Rouge'} 

men[:] = map(list,map(wom.intersection, men)) 

print(men) 

Вы могли бы также начать с конца списка, используя логику диапазона, но с использованием range(len(sub)-1,-1, -1), но проще всего использовать обратный и перебирать сами элементы:

def heroes_updater(men_pref, women): 
    for sub in men_pref: 
     for m in reversed(sub): 
      if m not in women: 
       sub.remove(m)