python在多个循环中迭代多个列表

Question

我需要遍历多个列表，并对匹配的记录做一些计算：

for (a,b,c,d) in list1:
   for (a2,b2,e) in list2:
       if (a==a2) and (b==b2):
           mylist.add(a,b,c,d,e,d*e)

是否有一种有效的方法来进行上述计算。非常感谢。

Answer 1

构建一些字典以便快速查找：

data1 = {(a, b): (c, d) for a, b, c, d in list1}
data2 = {(a, b): e for a, b, e in list2}

result = []
for a, b in set(data1) & set(data2):
    c, d = data1[a, b]
    e = data2[a, b]
    result.append((a, b, c, d, e, e*d))

Answer 2

牢记新信息，即：

1. list1 的元素格式为(name, last_name, gender, job_class, salary)，
2. list2 包含 (name, last_name, increase) 形式的元素（可能是对人的加薪），
3. list3 具有 (job_class, bonus) 等元素，

...使用dict，您可能会在性能和代码清晰度方面受益。

使用 (first,last) 形式的元组来引用程序中的每个人，您可以执行以下操作（在使用 input 获取信息的基本示例中）：

people = dict()
for i in range(num_ppl):
    name = tuple(input().split()) # input is something like "Bob Smith"
    people[name] = getPeopleInfo() # read gender, job_class, salary, etc. and make a list
for i in range(num_raises):
    first, last, increase = input().split()
    people[(first,last)][-1] *= float(increase)
for i in range(num_bonuses):
    job_class, bonus = input().split()
    for name in people: # iterating through a dict gives the keys (similar to indices of a list, but can be immutable types such as tuples)
        if people[name][2] == job_class:
            people[name][-1] += bonus

str、int 和tuple 等任何不可变类型都可以用作dict 中的键，类似于用于list 的基于0 的整数。请注意，list 可以更改（例如使用list.append）并且是“可变的”；因此 list 不能是键。有关dict 的更多信息，您可以阅读documentation。

Answer 3

就时间和内存效率而言，当前的代码似乎大多是最优的。您必须相互检查 list1 和 list2 的所有元素以进行比较。
消除一些重复“错误”情况的一种补充方法是在两个 for 循环行之间添加：

if a != b: # none of the items in list2 will satisfy a==b2 and b==b2
    continue

您也可以在 Python 中使用 if a == b == b2，而不必将语句与 and 绑定在一起。

根据您的记录的存储和访问方式，您可能会从使用dicts 而不是lists 中受益。 dict 可以判断是否有其实现的示例可能是：

lookup = dict()

# when adding an item to what would be list2
if b2 in not in lookup:
    lookup[b2] = []
lookup[b2].append((a2,e))
# ...

for (a,b,c,d) in list1:
    if a == b and a in lookup:
        for (a2,e) in lookup[a]:
            mylist.add(a,b,c,d,e,d*e)