python - 从 numpy 数组中删除无限值

Question

我想从一个数组B中删除与另一个数组中的元素具有相同索引的inf元素A。我有两个 numpy 数组，例如

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

如果我这样做B[A>2]，则输出为array([7, 8, 9]). 但是，如果我这样做B[math.isfinite(A)]，那么我会收到一个错误

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

如何选择B值A不是无穷大的元素？

标签： pythonarraysnumpy

Answer 1

我想你的问题有答案：

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