transform value as array

Question

I have this code:

import numpy as np

a = [0,np.array([1,2])]

try:
    # something like that
    [ x =  np.array([x]) for x in a if x == 0]
except ValueError:
    pass

I want to replace every zero value as an array, so my result would be:

a = [np.array([0]), np.array([1,2])

Hah @falsetru. Just noticed you got it first in the comment. — nir0s
– nir0s, Commented Mar 7, 2017 at 8:37

nir0s · Accepted Answer · 2017-03-07 08:34:32Z

2

Your list comprehension is not valid.

Use:

x = [np.array([x]) for x in a if x == 0]

instead.

Do note that the logic here doesn't provide the relevant answer but rather:

x = array([1, 2])

in the end.

For what you're expecting:

Use:

np.array([0]) if x is 0 else x for x in a

answered Mar 7, 2017 at 8:34

nir0s

1,2018 silver badges18 bronze badges

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

Collectives™ on Stack Overflow

transform value as array

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related