如何对有时间的python列表进行下采样

Question

我在 python 中有一个二维列表，其中包含图形的 X 和 Y 坐标。 像这样：

list1 = [[12345678, 12345678], [12345678, 12345678], [12345678, 12345678]]

每个内部列表的第一个元素是 X 轴，第二个是 Y 轴。所有 X 轴元素都是epoch (Unix timestamp)，我想以某种方式减少此列表中的元素数量，方法是仅在某个设定间隔（例如 5 分钟）后保留时间戳。因此，应该删除下午 1:00 到下午 1:05 之间的任何点...

那么，如何以最有效的方式做到这一点

Answer 1

如果我很好地理解了你的意思，那么这个代码示例可能会对你有所帮助。

import datetime
input_data = [[12345678, 12345678], [12346678, 12345678], [12346698, 12345678], [12446678, 12345678], [12455678, 12345678]]

# If you're sure that the input data is always sorted so you can skip this line input_data.sort(key=lambda x : x[0])

result = [input_data[0]] 
for i in range(1, len(input_data)):
    date = datetime.datetime.fromtimestamp(input_data[i][0])
    previous_date = datetime.datetime.fromtimestamp(result[-1][0])
    
    # 5 minutes = 5 * 60 = 300 seconds (you can change this to any time interval)
    if (date - previous_date).total_seconds() > 300 :
        result.append(input_data[i])

所以正如我上面提到的，如果您的列表已经排序，那么您可以跳过排序行。

那么这里发生了什么： 首先我们复制第一项 然后我们进入输入列表中的下一项，我们首先检查时差。如果差异小于 5 分钟，我们将忽略该项目，否则我们会将其附加到结果列表中。以此类推。

我希望这会有所帮助，否则你可以告诉我，如果我错过了理解你的问题。

---

PS：根据下面的评论，如果您使用枚举而不是范围，代码如下所示：

result = [input_data[0]]
for _, date_i in enumerate(input_data):
    date = datetime.datetime.fromtimestamp(date_i[0])
    previous_date = datetime.datetime.fromtimestamp(result[-1][0])

    # 5 minutes = 5 * 60 = 300 seconds (you can change this to any time interval)
    if (date - previous_date).total_seconds() > 300:
        result.append(date_i)