如何在Python 2.7中使用數組A的非無限數索引創建數組C？

Question

數組是這樣的：

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

列C應該是這樣的：

[0, 2, 3, 5] 

我怎樣才能讓列C與數組A的非無限的數字在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])