如何在Python 2.7中使用数组A的非无限数的索引来创建数组C？

Question

数组 A 看起来像这样：

[1, -inf, 2, 3, inf, -60.2]

数组 C 应该看起来像这样：

[0, 2, 3, 5]

How can I make array C with the non-infinite number of array A in Python 2.7?

Answer 1

inf = float('inf')
a = [1, -inf, 2, 3, inf, -60.2]
c = [i for i, v in enumerate(a) if abs(v) != inf]

Answer 2

如果“数组”是指“numpy 数组”：

In [12]: import numpy as np

In [13]: a = np.array([1, -np.inf, 2, 3, np.inf, -60.2])

In [14]: np.where(~np.isinf(a))[0]
Out[14]: array([0, 2, 3, 5])