Variable Size Sliding Window Algorithm Python