python - 如何获取元素所在的列表？

Question

假设我有两个列表:

L1 = [a, b, c, d]

和

L2 = [e, f, g, h]

字母可以是任何类型(没有区别)。

有没有办法问 Python:“哪个列表(L1 和/或 L2)包含元素 'a'(或任何其他元素)，并且具有它返回该列表的名称？

Answer 1

为了实现这一点，更好的数据结构将使用 dict .例如:

my_list_dict = {
    'L1': ['a', 'b', 'c', 'd'],
    'L2': ['e', 'f', 'g', 'h']
}

然后您可以创建一个自定义函数来完成您的任务:

def find_key(c):
    for k, v in my_list_dict.items():
        if c in v:
            return k
    else:
        raise Exception("value '{}' not found'.format(c))

样本运行:

>>> find_key('c')
'L1'
>>> find_key('g')
'L2'