通过多个维度索引列表的方法

Question

我正在编写使用嵌套列表的代码。我的列表结构本质上是 2 个元素，但第二个元素包含两个单独的元素，依此类推。因此，列表的大小增长为 2^n + 1。无论如何，对于函数的每次递归，我需要能够访问某个 'depth' 处的任何给定元素。通常，这将是微不足道的，

list1 = [[0,1], [[1,2], [1,0]]]

以下代码：

list1[1][1]

会 return:

[1,0]

但是，对于 n 维嵌套列表（我使用 'dimension' 这个词相当随意，因为这是一个嵌套列表而不是数组）我肯定需要 'exponentiate' 我的 [1 ] 索引以便索引到每个 'deeper' 嵌套列表中。有没有简单的方法可以做到这一点？

在此先致谢，非常感谢您的帮助。

Answer 1

使用递归：

def get(li, k):
    return get(li[1], k-1) if k > 0 else li

或迭代：

def get(li, k):
    for _ in range(k):
        li = li[1]
    return li

Answer 2

我创建了这个，我试图让它尽可能地不言自明：

my_list = [[8,9], [[0,1], [[1,2], [1,0]]]]
# dim=1 -> [8,9]
# dim=2 -> [0,1]
# dim=3 -> [1,2]
# dim=4 -> [1,0]

def get_value_by_dimension(dim :int, lst: list):
    x = dim-1
    if x==0:
        return (lst[0])
    else:
        try:
            return get_value_by_dimension(x, lst[1])
        except:
            return "Dim is to high!"
print(get_value_by_dimension(2, my_list))
# [0, 1]

print(get_value_by_dimension(1, my_list))
# [8, 9]

print(get_value_by_dimension(5, my_list))
# Dim is to high!

通过多个维度索引列表的方法

Method for indexing into a list by a number of dimensions

python

indexing

list

nested-lists