python - 获取两个数组中匹配元素的值

Question

如果 a 和 b 是对象列表，每个对象都有一个名称属性(例如 a1 = A("1")，b1 = B("1") 等)，我将如何检查等效性？我目前正在这样做:

aList = [ a1, a2, a3, a4, a5 ]
bList = [ b1, b2 ]
aNameList = []
bNameList = []

for i in aList:
    aNameList.append( i.name )
for i in bList:
    bNameList.append( i.name )

match = set(aNameList) & set(bNameList)

>>> set(['1', '2'])

但它似乎有点长而且没有必要。执行此操作的更好方法是什么？

Answer 1

您可以使用列表推导来替换示例中的那些临时列表和 for 循环:

match = set( [ x.name for x in aList ] ) & set ( [ x.name for x in bList ] )