Python: Compare array elements with float, getting a boolean list

Question

I want to compare all elements of a two-dimensional list array with a float x. The result should be a list: b = [[True, False,...],...]. I tried it like that:

import numpy as np

array = [[a1,a2,...], [a3,a4,...],...,] 
x = 2.0

b = np.array([a >= x for a in array])`

"TypeError: '>=' not supported between instances of 'list' and 'float'"

When I use a one-dimensional list it works fine.

Thanks in advance!!

array = np.array([[a1,a2,...], [a3,a4,...],...,] ) and if you still get the error, that means your list elements are ragged (irregular in size). — cs95
– cs95, Commented Apr 24, 2018 at 18:43
In you code b = np.array([a >= x for a in array]), since array is multidimensional, a is a list. So in this case you are comparing a (a list) with x (a float). — Patrick
– Patrick, Commented Apr 24, 2018 at 18:47
Can you provide an example of a portion of array, say array[:2,:2]? — wwii
– wwii, Commented Apr 24, 2018 at 18:54

Jan K · Accepted Answer · 2018-04-24 18:45:41Z

1

b = np.array([[a >= x for a in row] for row in array])

answered Apr 24, 2018 at 18:45

Jan K

4,1501 gold badge18 silver badges17 bronze badges

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

Collectives™ on Stack Overflow

Python: Compare array elements with float, getting a boolean list

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