Как я могу сделать массив C с индексами бесконечных чисел массива A в Python 2.7?

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

Если под «массивом» вы подразумеваете «массивный массив»:

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