python - Python 中的交叉引用列表

Question

我正在尝试创建一个函数，它接受一个输入，将该输入与一个列表列表进行比较，然后从另一个列表中返回一个具有相同数量对象的项目。

例子:

list_1=[[1,2,3],[4,5],[6,7,8]]

list_2=['a','b','c']

• 如果 input 为 1、2 或 3，函数返回 'a'

• 如果 input 是 4 或 5 函数返回 'b'

• 如果 input 是 6、7 或 8 函数返回 'c'

我是 python 的新手，一直在思考这个问题一段时间，四处寻找线索，但没有结果。任何可能帮助我解决这个问题的提示/线索将不胜感激!谢谢!

Answer 1

您可以遍历 list_1 中的每个列表并检查输入是否在一个列表中。如果是，则可以打印 list_2 的相应索引(假设它仅由单个值组成)，这是通过在循环中使用 enumerate 获得的。

input = 1
for idx,i in enumerate(list_1):
    if input in i:
        return list_2[idx]

在这种情况下，我返回了 'a'。