Python - List comphrehension with test to avoid division by zero

Question

a and b are 1 dimensional numpy arrays (or python lists):

I am doing this:

>>> c = [x/y for x,y in zip(a,b)]

Occasionally b has a zero in it - so a division by zero error occurs.

How can I conditionally check for a 0 value in b and set the corresponding element of c to 0?

Answer 1

You can use if-else condition inside list comprehension:

>>> c = [x/y if y else 0 for x,y in zip(a,b)]

Answer 2

You can use a ternary expression inside the list comprehension:

[x/y if y!= 0 else 0 for x,y in zip(a,b)]