如何用单个优雅的Python列表推导式提取支点关联点？

需求与现有实现 #

我们需要从一组表示点间连接的索引对列表中，提取所有与指定支点相连的节点索引——无论支点在索引对的第一个还是第二个位置，只需取出对中的另一个节点。

现有实现是通过两个列表推导式拼接：

bonds = [(1,2),(3,4),(5,6),(3,1)]
ipiv  = 1 

bonded_to_pivot = [b[1] for b in bonds if(b[0]==ipiv)] + [b[0] for b in bonds if(b[1]==ipiv)]

单个列表推导式的实现方案 #

当然可以用单个优雅的列表推导式完成，这里提供两种思路：

方案一：条件判断取对应元素

bonds = [(1,2),(3,4),(5,6),(3,1)]
ipiv = 1

bonded_to_pivot = [b[1] if b[0] == ipiv else b[0] for b in bonds if ipiv in b]

逻辑说明：

• 先通过if ipiv in b过滤出所有包含支点的索引对
• 对每个符合条件的索引对，判断支点是否在第一个位置：是则取第二个元素，否则取第一个元素

方案二：嵌套遍历筛选元素

bonds = [(1,2),(3,4),(5,6),(3,1)]
ipiv = 1

bonded_to_pivot = [x for b in bonds for x in b if x != ipiv and ipiv in b]

逻辑说明：

• 遍历每个索引对，再遍历对中的每个元素
• 筛选出不等于支点且所在索引对包含支点的元素，恰好就是所有与支点相连的节点

这两种方式都能达到原代码的效果，且只用了单个列表推导式，更简洁紧凑。

内容的提问来源于stack exchange，提问作者Prokop Hapala