Python: Сравнение списков

Question

Я столкнулся с небольшой проблемой. Скажем, у меня есть два списка:

list_A = ['0','1','2'] 
list_B = ['2','0','1'] 

Я тогда есть список списков:

matrix = [ 
['56','23','4'], 
['45','5','67'], 
['1','52','22'] 
] 

Я тогда нужно перебрать list_A и list_B и эффективно использовать их в качестве координат. Например, я беру первое число из списка A и B, которое будет «0» и «2», затем я использую их как координаты: print matrix[0][2]

Мне тогда нужно сделать то же самое для второго номера в списке A и B и 3-е число в списках A и B и т. Д. Для любого длинного списка A и B, как бы это было. Как это делается в цикле?

Answer 1

matrix = [ 
['56','23','4'], 
['45','5','67'], 
['1','52','22'] 
] 

list_A = ['0','1','2'] 
list_B = ['2','0','1'] 

for x in zip(list_A,list_B): 
    a,b=map(int,x) 
    print(matrix[a][b]) 
# 4 
# 45 
# 52 

Answer 2

Здесь может быть использована функция «zip». Он будет генерировать список пар из list_A и list_B.

for (x,y) in zip(list_A, list_B): 
    # do something with the coordinates 

Answer 3

[matrix[int(a)][int(b)] for (a,b) in zip(list_A, list_B)]