python - Python 2.7で配列Aの非無限数のインデックスを使用して配列Cを作成するにはどうすればよいですか?

Question

配列 A は次のようになります。

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

配列 C は次のようになります。

[0, 2, 3, 5]

Python 2.7で配列Aの非無限数のインデックスを使用して配列Cを作成するにはどうすればよいですか?

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])