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?

Does [this](https://stackoverflow.com/questions/17477979/dropping-infinite-values-from-dataframes-in-pandas) answer your question? — Ivan, Feb 03 '20 at 10:52
I am using numpy arrays, not pandas dataframes. The methods aren't quite the same — usernumber, Feb 03 '20 at 10:53
Also, I the question isn't about dropping `inf`s from `A`, but rather about dropping elements in `B` that are in the same position as `inf`s from `A`. — usernumber, Feb 03 '20 at 10:54

score 2 · Accepted Answer · answered Feb 03 '20 at 10:57

2

I think you have the answer in your question:

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

answered Feb 03 '20 at 10:57

Ivan

1,352
2
13
31

Remove infinite values from numpy array

1 Answers1