numpy.ndarray.strides

ndarray.strides

Кортеж байтов для перемещения по каждому измерению при обходе массива.

Смещение в байтах элемента (i[0], i[1], ..., i[n]) в массиве a:

offset = sum(np.array(i) * a.strides)

Более подробное объяснение шагов можно найти в файле «ndarray.rst» в справочном руководстве NumPy.

См. также

numpy.lib.stride_tricks.as_strided

Примечания

Представьте массив 32-битных целых чисел (каждое число занимает 4 байта):

x = np.array([[0, 1, 2, 3, 4],
              [5, 6, 7, 8, 9]], dtype=np.int32)

Этот массив хранится в памяти как 40 байтов, один за другим (известен как непрерывный блок памяти). Шаги массива показывают, сколько байтов нужно пропустить в памяти, чтобы перейти к следующей позиции вдоль определённой оси. Например, необходимо пропустить 4 байта (1 значение), чтобы перейти к следующему столбцу, но 20 байтов (5 значений), чтобы попасть в ту же позицию в следующей строке. Таким образом, шаги для массива x будут (20, 4).

Примеры

>>> y = np.reshape(np.arange(2*3*4), (2,3,4))
>>> y
array([[[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]],
       [[12, 13, 14, 15],
        [16, 17, 18, 19],
        [20, 21, 22, 23]]])
>>> y.strides
(48, 16, 4)
>>> y[1,1,1]
17
>>> offset=sum(y.strides * np.array((1,1,1)))
>>> offset/y.itemsize
17

>>> x = np.reshape(np.arange(5*6*7*8), (5,6,7,8)).transpose(2,3,1,0)
>>> x.strides
(32, 4, 224, 1344)
>>> i = np.array([3,5,2,2])
>>> offset = sum(i * x.strides)
>>> x[3,5,2,2]
813
>>> offset / x.itemsize
813