python - numpy(.ma) 数组 : Number of values since last value change?

Question

我在 numpy.ma 数组中有一个状态信号(测量的，热泵的)，连同时间戳。我想要的是它打开的时间长度和它关闭的时间长度。 (不是每日运行时间之类的，那很容易..)

我有什么(示例性的，事实上我有超过 16 个月的分钟值..):

Time    Value
12:00   0
12:01   1
12:02   1
...
12:29   1
12:30   1
12:31   0
...
12:41   0
12:42   1
...
13:01   1
13:02   0
...and so on

以及我想要的输出:

running_time_was:
time   value (=minutes)
12:31  30
13:10  20

off_time_was:
time   value (=minutes)
12:42  11

(the exact timestamps doesn't matter so much, but there should be some)

我已经问过我认识的懂 python 的人(他们也不知道)，并尝试在互联网上搜索，但我只是不知道要搜索什么。所以也许有人至少可以给我一个提示，我可以在谷歌中输入哪些词？ :)

ps:哇，stackoverflow 太棒了!我已经对作为被动用户的可用性感到惊讶，但询问界面甚至更好:)

Answer 1

好吧，我终于得到了我自己。但是非常感谢 Joe Kington 的回答，它极大地启发了我并教会了我所有的东西:)

脚本

import numpy as np

def main():
    # Generate some data...
    t = np.linspace(0, 10*np.pi, 30)
    x = np.sin(t)
    condition = np.where(x>0,1,0)
    
    onarray,offarray = on_off_times(condition)

    print "Condition: ",condition
    print "Ontimes:   ",onarray
    print "Offtimes:  ",offarray
    
        
def on_off_times(condition):
        
    changing=np.diff(condition)         #complete array, -1 when tunring off, +1 when turning on
    idx, = changing.nonzero()           #Indices of changepoints
    times=np.diff(idx)              #'Times' between changes
    times=np.r_[0,times]            # The first time can't be calculated ->is set to 0
    
    ontimes=np.where(changing[idx]<0,times,False)   #When turning off: Was Ontime (times-array with "False" instead of offtimes)
    offtimes=np.where(changing[idx]>0,times,False)  #When turning on:  Was Offtime
    
    onarray=np.r_[changing.copy(),0]            #copy the array with all the values and add 1(to have an "empty" array of the right size)
    offarray=np.r_[changing.copy(),0]
    
    np.put(onarray,idx,ontimes)         #Put the times into it (at the changepoints)
    np.put(offarray,idx,offtimes)

    return onarray,offarray
    
main()

产量:

Condition:  [0 1 1 0 0 0 1 1 1 0 0 0 1 1 1 0 0 0 1 1 1 0 0 0 1 1 1 0 0 0]
Ontimes:    [0 0 2 0 0 0 0 0 3 0 0 0 0 0 3 0 0 0 0 0 3 0 0 0 0 0 3 0 0 0]
Offtimes:   [0 0 0 0 0 3 0 0 0 0 0 3 0 0 0 0 0 3 0 0 0 0 0 3 0 0 0 0 0 0]

要提及的事情:

• 是的，代码给出了我在问题中要求的另一个输出。对不起，不精确，但函数给出的输出是什么我终于需要了。
• 现在我只需要在输出中得到零蒙面，这很容易。
• 如果有人有更好的方法将数据放入新数组，请编辑 :)
• 如何链接到用户 Joe？或者如何感谢他？