Кодирование массива

In[1]: encode(np.array([1, 2, 2, 3, 3, 1, 1, 5, 5, 2, 3, 3])) 
Out[1]: (np.array([1, 2, 3, 1, 5, 2, 3]), np.array([1, 2, 2, 2, 2, 1, 2]))

import numpy as np
 
def encode(arr):
    # Получаем индексы, где значение элемента меняется
    indices = np.where(arr[:-1] != arr[1:])[0]
 
    # Добавляем индекс последнего элемента
    indices = np.concatenate((indices, [len(arr) - 1]))
 
    # Разбиваем массив на части, используя индексы
    parts = np.split(arr, indices + 1)
 
    # Получаем части, состоящие из одного элемента
    values, counts = np.unique(parts, return_counts=True)
 
    return values, counts

def encode(arr):
indices = np.where(arr[:-1] != arr[1:])[0]
indices = np.concatenate((indices, [len(arr) - 1]))
parts = np.split(arr, indices + 1)
values, counts = np.unique(parts, return_counts=True)
return values, counts

import numpy as np
def encode(arr):
    tmp = np.r_[True,arr[:-1]!=arr[1:],True]
    return (arr[tmp[:-1]], np.diff(np.flatnonzero(tmp)))
 
 
a = np.array([1, 2, 2, 3, 3, 1, 1, 5, 5, 2, 3, 3])
print(encode(a))

from functools import reduce
from random import randint as rndi
from itertools import groupby
 
GROUP = 10_000
encode = tuple(reduce(lambda x, y: x + y, [[rndi(1, 10)] * rndi(1, 10) for _ in range(GROUP)], []))

def groupby_encode(a):
    return [x for x, y in groupby(a)], [len(list(y)) for x, y in groupby(a)]

[(x,list(y)) for x, y in groupby(arr)]

import numpy as np
 
def encode_func1(a):
    a = np.asarray(a)
    v = np.hstack(((a[:-1] != a[1:]), True))  
    z = np.nonzero(np.hstack((True, v)))      
    return np.array(a[v]), np.diff(z)[0]

def encode_func2(a):
    v = np.hstack(((a[:-1] != a[1:]), True))  
    z = np.nonzero(np.hstack((True, v)))      
    return np.array(a[v]), np.diff(z)[0]