根据来自其他 2 个列表的索引比较 2 个列表，并使用列表理解将索引保存在新列表中

Question

So what I am trying to do is:

1) Find where list1[y] == list2[x]

2) Determine if list3[y] < (0.4 list4[x])

3) If so, store the index y in a new list of indexes

This loop works, however it takes almost a full minute to run with my data set. I want to know if I can do this with list comprehension.

Thanks in advance for anybody's help.

    for y in range(len(list1)):
        for x in range(len(list2)):
            if list1[y] == list2[x]:
                if list3[y] < (0.4 * list4[x]):
                    list5.append(y)

Answer 1

我不确定这是否会显着提高性能，但试试这个：

list5 = [y for y, val1 in enumerate(list1) for x, val2 in enumerate(list2) if val1 == val2 and list3[y] < (0.4 * list4[x])]