在另一个列表中找到每个子列表的超集索引

时间:2018-12-19 10:09:41

标签: python list indexing set

我试图在与另一个列表中的元素共享相同数据的元素列表中找到索引号。这些是我的清单:

list_A = [['A',1,'a',],['B',2,'b'],['C',3,'c'],['D',4,'d'],['E',5,'e'],['F',6,'f']]
list_B = [['A','a'],['D','d']['E','e']]

所需的输出应为:

[0,3,4]

我已经尝试过使用set()交集和index函数,但是无法对列表列表进行'set'操作。

2 个答案:

答案 0 :(得分:4)

如果列表值没有重复项,则可以将所有内容转换为集合,并在列表推导内执行子集检查。

set_A = list(map(set, list_A))
set_B = list(map(set, list_B)) 
# set_B = map(set, list_B) # If you intend to use and throw. 

[next((i for i, a in enumerate(set_A) if a.issuperset(b)), None) for b in set_B]
# [0, 3, 4]

现在,如果没有匹配项,next返回您传递给它的默认值,在这种情况下为None

答案 1 :(得分:1)

您可以执行以下操作:

list_A = [['A', 1, 'a', ], ['B', 2, 'b'], ['C', 3, 'c'], ['D', 4, 'd'], ['E', 5, 'e'], ['F', 6, 'f']]
list_B = [['A', 'a'], ['D', 'd'], ['E', 'e']]

set_B = list(map(set, list_B))
result = [i for i, e in enumerate(list_A) if any(b.intersection(e) for b in set_B)]

print(result)

输出

[0, 3, 4]