python:compare if string in list

Question

I have a problem.
I have an product.address like this 'UK some city some road'

And I want to compare this address,if it is start with UK or US or AS or GE
I will save product.dstore_id = 2,otherwise, save product.dstore_id = 1

My code is like this:

area = [UK,US,AS,GE]
for i in area:
    if i in product.address:
        product.dstore_id = 2
    else:
        product.dstore_id = 1

But I found this has a big problem,it will compare all the i and save the wrong number
So how can I do to reach it . Please guide me a bit Thank you

Cory Kramer · Accepted Answer · 2014-10-13 13:13:02Z

4

You could use any. This will return True if any of the items in your area list are present in the product.address. You then no longer need the for loop.

if any(address.startswith(i) for i in area):
    product.dstore_id = 2
else:
    product.dstore_id = 1

For example

area = ['UK','US','AS','GE']
address =  'UK some city some road'
>>> any(addres.startswith(i) for i in area)
True

edited Oct 13, 2014 at 13:13

answered Oct 13, 2014 at 12:07

Cory Kramer

119k19 gold badges176 silver badges233 bronze badges

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

1 Comment

PM 2Ring Over a year ago

startswith is safer & more efficient, eg product.dstore_id = 2 if any(address.startswith(i) for i in area) else 1

Collectives™ on Stack Overflow

python:compare if string in list

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related