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