Remove infinite values from numpy array

Question

I would like to remove elements from one array B that have the same index as the inf elements from another array A. I have two numpy array such as

A = np.array([1,2,3,4, float('inf')])
B = np.array([5, 6, 7, 8, 9])

If I do B[A>2], the output is array([7, 8, 9]). However, if I do B[math.isfinite(A)], then I get an error

TypeError: only size-1 arrays can be converted to Python scalars

How can I select the elements from B where the value in A is not infinity?

I am using numpy arrays, not pandas dataframes. The methods aren't quite the same — usernumber
– usernumber, Commented Feb 3, 2020 at 10:53
Also, I the question isn't about dropping infs from A, but rather about dropping elements in B that are in the same position as infs from A. — usernumber
– usernumber, Commented Feb 3, 2020 at 10:54

Ivan · Accepted Answer · 2020-02-03 10:57:32Z

2

I think you have the answer in your question:

B = B[A!= float('inf')]

answered Feb 3, 2020 at 10:57

Ivan

1,5512 gold badges20 silver badges42 bronze badges

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

Collectives™ on Stack Overflow

Remove infinite values from numpy array

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related