Python2: доступ к вложенному списку с использованием списка индексов

Question

Как получить доступ к элементу вложенного списка с другим списком, который содержит индексы?

например:

# this is the variable containing the indices 
a = [0, 1] 

b = [[1,2],[3,4]] 

в действительности, эти списки заполнены элементами самоуправления определены классы и список, содержащий «координаты» (а) имеет более 2-х элементов.

Есть ли возможность получить доступ к b [0] [1] автоматически? Раньше я использовал этот код:

c = deepcopy(b) 
for el in a: 
    c = c[el] 

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

Я доволен любыми предложениями :)

Спасибо!

Answer 1

Просто бросьте его в функцию. Это будет держать его область действия, так что вы не перезаписывать исходное значение

def nested_getitem(container, idxs): 
    haystack = container 
    for idx in idxs: 
     haystack = haystack[idx] 
    return haystack 

DEMO:

>>> a = [0, 1] 
>>> b = [[1, 2], [3, 4]] 
>>> nested_getitem(b, a) 
2 

Вы могли бы сделать это с functools.reduce, а также, если вы были безумны.

import functools 
import operator 

def nested_getitem(container, idxs): 
    return functools.reduce(operator.getitem, idxs, container)