如何用Python列表推导式实现带条件的嵌套循环逻辑？

问题回顾 #

你需要判断l1中的每个子列表是否在l2的子列表前两项中存在，存在则标记为1，否则为0，预期输出[0,1,1]，但当前的列表推导式生成了二维列表，不符合需求。

核心逻辑分析 #

原循环的本质是：对l1的每个元素，检查是否存在至少一个l2的元素，其前两项与当前l1元素完全匹配，匹配则记1，否则记0。

解决方案 #

方法1：用集合优化查找（推荐，效率更高） #

先把l2的前两项提取为集合（集合的查找时间复杂度是O(1)），再遍历l1判断元素是否在集合中：

l1 = [['a',1],['b',2],['c',3]]
l2 = [['b',2,10],['c',3,8]]

# 将l2的前两项转为元组存入集合（列表不能作为集合元素，元组可以）
l2_pairs = {(item[0], item[1]) for item in l2}

# 列表推导式实现逻辑
isintb = [1 if (x[0], x[1]) in l2_pairs else 0 for x in l1]
print(isintb)  # 输出: [0,1,1]

方法2：直接用any()函数判断匹配 #

如果不需要优化效率，也可以直接在推导式中用any()判断是否存在匹配项：

l1 = [['a',1],['b',2],['c',3]]
l2 = [['b',2,10],['c',3,8]]

isintb = [1 if any(x[0] == y[0] and x[1] == y[1] for y in l2) else 0 for x in l1]
print(isintb)  # 输出: [0,1,1]

为什么你的写法不对？ #

你原来的推导式：

[[1 if (l1[i][0]==l2[j][0]) & (l1[i][1]==l2[j][1]) else 0 for j in range(len(l2))] for i in range(len(l1))]

是对每个i（l1的元素），生成一个包含所有j（l2的元素）匹配结果的列表，所以得到的是二维数组[[0, 0], [1, 0], [0, 1]]。而我们需要的是对每个i，判断这个内层列表中是否有至少一个1，any()函数正好可以完成这个判断，再转成int就是需要的0/1。

另外注意：原代码中的&是位运算符，逻辑判断建议用and，语义更清晰。

内容的提问来源于stack exchange，提问作者Revanth P S