熊貓 - 加速計算

Question

我有一個數據框，'數據'如下：

data.head(3)
     Time          Value
0    01:00:00.125  5
1    01:02:03.145  3
2    01:02:03.651  9

我正在嘗試創建一個新列，在每一行上提供現有行的最高“值”以及該行之前在當前行的時間戳之前5秒或更短時間內發生的所有行。 我目前的方法是一個非常慢的循環，我想知道是否有更快的方法？

作為參考，我當前的代碼循環遍歷所有行，獲取當前時間，計算當前時間之前5秒的時間，創建一個僅包含該時間間隔內的行的新幀，然后取最大值價值'專欄。

Answer 1

您可以在pandas中使用滾動功能：

col = data.rolling(window='5s', on='Time').max()
data['aggregate'] = col['Value']