Python 将区间以 [1,0,1] 开始和停止的列表转换为阶跃函数 [0, 1]

发布于07月01日

假设给一个人一个列表，表示事件发生的时间.事件的开始标记为1，事件的结束标记为-1.否则，对于每个采样的时间点，它都是0.举个简单的例子:

event_list = [0,0,0,0,1,0,0,-1,0,0,0,0,0,0,0,0,1,0,0,0,0,0,-1,0,0,0,0,0,1,0,0,0,0,0,-1,0,0]

我想找到一个简单的方法，将该列表转换为步长函数，其中事件的开始由1表示，事件的持续时间都是1，当事件结束时，它返回到0.因此，对于与上面相同的列表，它看起来像:

step_list = [0,0,0,0,1,1,1,1,0,0,0,0,0,0,0,0,1,1,1,1,1,1,1,0,0,0,0,0,1,1,1,1,1,1,1,0,0]

现在，可以通过在列表上循环并使用if语句和缓冲区来创建新列表来强制执行此操作.但必须有一种比这更优雅的方式.我想在numpy中使用diff函数，但这仍然不太管用.谢谢你的建议！

推荐答案

可能有一种更优雅的方法(numpy可能会令人惊讶)，但您可以通过两个步骤来实现这一点，即.cumsum()和索引替换:

import numpy as np
x = np.array(event_list)

result = x.cumsum()
result[x == -1] = 1

输出:

>>> np.array_equal(result, step_list)
True

这是通过首先获取元素的累积和来实现的.遇到的第一个1将总和设置为1.它后面跟着零(这意味着总和保持在1)，直到遇到-1，然后总和"重置"回0(直到遇到下一个1).这本质上是将1"向前填充"到-1，但不包括-1，因此最后一步是简单地将1分配给索引，其中原始数组的值为-1.

实用课程推荐