我有两个列表列表
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')]]
最佳答案
您似乎没有使用所有元素。但是,我会从 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]]
(请注意,携带其他未使用的元素并不能真正帮助理解问题)
关于python - 将一个嵌套列表的元素与另一个嵌套列表进行比较,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47928029/