python - 将一个嵌套列表的元素与另一个嵌套列表进行比较

Question

我有两个列表列表

l1 = [[1,2,3],[4,5,6],[7,8,9]]
l2 = [['a','b',4],['c','d',1],['e','f',12],['i','j',18]]

我想遍历 l1并检查是否l1[0]匹配任何 l2[2] , 在这种情况下输出应该是 [1, l1[0],l2[0]]否则输出为 [0, l1[0], l2[0]] .输出应该是单个嵌套列表(或元组列表)，其中包含 l1 的每个元素的结果。 .两个列表可以有不同的大小。

我试着用 for-loop 来解决这个问题，比如:

output = list()
for i in l1:
   matched = 0
   for j in l2:
       if j[2] == i[0]:
          output.append([1,i[0], j[0]])
          matched = 1
    if matched == 0:
       output.append([0,i[0]])

这给出了正确的输出

[[1, 1, 'c'], [1, 4, 'a'], [0, 7]]

但是我正在寻找更紧凑的解决方案。是否可以通过对类似内容的列表理解来解决此问题，从而减少所涉及的行数？

我尝试了嵌套列表推导式，但无法使其工作

out = [[(1,i[0],k[0]) if(k[2] == i[0]) else (0,i[0],k[0]) for k in l2] for i in l1]
print(out)
[[(0, 1, 'a'), (1, 1, 'c'), (0, 1, 'e'), (0, 1, 'i')], [(1, 4, 'a'), (0, 4, 'c'), (0, 4, 'e'), (0, 4, 'i')], [(0, 7, 'a'), (0, 7, 'c'), (0, 7, 'e'), (0, 7, 'i')]]

Answer 1

您似乎没有使用所有元素。但是，我会从 l2 中构建一个 dict 以便快速查找和简洁(一行可能是可能的，但以牺牲可读性和性能为代价)

我会在后面加上一个列表理解，包括一个三元组，根据是否找到该项目来发出 2 或 3 个元素(因此不需要花哨的 int(if a in l2d) 因为我们可以直接发出 0 或 1)。像这样:

l1 = [[1,2,3],[4,5,6],[7,8,9]]
l2 = [['a','b',4],['c','d',1],['e','f',12],['i','j',18]]

l2d = {v[2]:v[0] for v in l2}  # not using v[1]

result = [[1,a,l2d[a]] if a in l2d else [0,a] for a,_,_ in l1]  # using only first element of each l1 triplet...

结果:

[[1, 1, 'c'], [1, 4, 'a'], [0, 7]]

(请注意，携带其他未使用的元素并不能真正帮助理解问题)