Question

我具有生成二进制信号的时间序列，如下所示：

date    value
1/4/1987    0
1/5/1987    1
1/6/1987    1
1/7/1987    0
1/9/1987    0
1/10/1987   1
1/12/1987   1
2/1/1987    1
2/2/1987    1
2/3/1987    1
2/4/1987    1
2/6/1987    1
2/7/1987    1
2/9/1987    0
2/10/1987   0
2/11/1987   0
2/12/1987   1
3/2/1987    0
3/3/1987    1
3/4/1987    1
3/6/1987    1
3/8/1987    1
3/9/1987    1
3/11/1987   1
3/12/1987   0

我正在尝试找出减少它们数量的方法，以便在1个之间固定固定10个观察值。

date    new_value
1/4/1987    0
1/5/1987    1
1/6/1987    0
1/7/1987    0
1/9/1987    0
1/10/1987   0
1/12/1987   0
2/1/1987    0
2/2/1987    0
2/3/1987    0
2/4/1987    0
2/6/1987    0
2/7/1987    1
2/9/1987    0
2/10/198    0
2/11/198    0
2/12/198    0
3/2/1987    0
3/3/1987    0
3/4/1987    0
3/6/1987    0
3/8/1987    0
3/9/1987    0
3/11/1987   1
3/12/1987   0

任何帮助深表感谢。

Answer 1

想出矢量化的方法来完成这项工作似乎很棘手。在这些情况下，如果我们仍希望使用高性能方法，那么numba是一个不错的选择。使用其有效的@njit编译模式的方法如下：

from numba import njit
import numpy as np

@njit
def spacing_between_1(a, k):
    x = np.zeros(len(a), np.int8)
    first_one = np.argmax(a)
    x[first_one] = 1
    c=0
    for i in range(first_one+1, len(x)):
        if a[i] == 1 and c >= k:
            x[i] = 1
            c=0
            continue
        c +=1
    return x

对于共享示例，我们将得到：

a = df.value.to_numpy()
df['new_value'] = spacing_between_1(a, 10)

print(df)

           value  new_value
date                       
1/4/1987       0          0
1/5/1987       1          1
1/6/1987       1          0
1/7/1987       0          0
1/9/1987       0          0
1/10/1987      1          0
1/12/1987      1          0
2/1/1987       1          0
2/2/1987       1          0
2/3/1987       1          0
2/4/1987       1          0
2/6/1987       1          0
2/7/1987       1          1
2/9/1987       0          0
2/10/1987      0          0
2/11/1987      0          0
2/12/1987      1          0
3/2/1987       0          0
3/3/1987       1          0
3/4/1987       1          0
3/6/1987       1          0
3/8/1987       1          0
3/9/1987       1          0
3/11/1987      1          1
3/12/1987      0          0

强制二进制数组之间的最小分隔

1 个答案: