2011-07-27 5 views
0

배열 A는 다음과 같습니다파이썬 2.7에서 무한 개수의 배열 A의 인덱스로 배열 C를 만들려면 어떻게해야합니까?

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

배열 C는 다음과 같이한다 :

[0, 2, 3, 5] 

어떻게 파이썬 2.7에서 배열 A의 비 무한한 숫자의 인덱스와 배열 C를 만들 수 있습니까?

+1

매우 가까운 (http://stackoverflow.com/questions/6841837/how-를 can-i-make-array-b-from-array-a-without-without-infinities-in-python-2-7)을 사용하면됩니다. 팁 :'enumerate'를보십시오. –

답변

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

"배열"에 의한 경우 당신은 "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])