Python:替换numpy数组的每第5个值

Question

我有一个充满 float 的 numpy 数组。如何用 np.inf*0 替换每第 5 个值，以便在每第 5 个索引处得到一个 NaN 值？

my_array = np.array([5.0, 8.1, 3.2, 2.7, 8.4, 4.9 ...])

到

my_array = np.array([5.0, 8.1, 3.2, 2.7, NaN, 4.9 ...])

等等。

Answer 1

使用切片和跨步怎么样？ L[::5] 从列表 L 中每隔 5 个元素取一次:

>>> my_array = np.arange(20.)
>>> my_array[4::5] = np.nan
>>> my_array
array([  0.,   1.,   2.,   3.,  nan,   5.,   6.,   7.,   8.,  nan,  10.,
        11.,  12.,  13.,  nan,  15.,  16.,  17.,  18.,  nan])