optimizing looping in python in list operation

Question

I am using the following code for the iteration. but if the length is high, then this will become time-consuming.

a=[11,12,3,4,5,6,15]
b=[1,3,12,15]
c=[1,2,3,4,5,6,7]
d=[]
for i in range(len(a)):
    for j in range(len(b)):
        if b[j]==a[i]:
            d.append(c[i])
print(d)

is there any other optimal way to get this done?

thanks in advance

Making a dict of b will help considerably.

sj95126
– sj95126

2021-12-23 01:05:41 +00:00
Commented Dec 23, 2021 at 1:05 — sj95126
– sj95126, Commented Dec 23, 2021 at 1:05

yjay · Accepted Answer · 2021-12-23 01:07:50Z

3

Checking if collection contains an object is optimized for sets. You could do:

b_set = set(b)
[ci for ai, ci in zip(a, c) if ai in b_set]

answered Dec 23, 2021 at 1:07

yjay

1,0235 silver badges14 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Pedro Maia · Accepted Answer · 2021-12-23 01:06:22Z

2

Use in:

for i,v in enumerate(a):
    if v in b:
        d.append(c[i])

Or a list comprehension:

d = [v for i,v in enumerate(c) if a[i] in b]

answered Dec 23, 2021 at 1:06

Pedro Maia

2,7321 gold badge7 silver badges21 bronze badges

Collectives™ on Stack Overflow

optimizing looping in python in list operation

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related