Как сдвинуть двумерный массив в Python? - PullRequest
0 голосов
/ 21 июня 2019

Как можно сместить элементы двумерного массива в Python?

Например, преобразовать этот массив:

[[ 1, 2, 3],
[ 4, 5, 6],
[ 7, 8, 9],
[10, 11, 12]]

к этому массиву:

[[0, 0, 0,],
[ 1, 2, 3],
[ 4, 5, 6],
[ 7, 8, 9]]

Ответы [ 2 ]

1 голос
/ 21 июня 2019
import numpy as np
a = np.array([[ 1, 2, 3],
              [ 4, 5, 6],
              [ 7, 8, 9],
              [10, 11, 12]])    

def shift_array(array, place):
    new_arr = np.roll(array, place, axis=0)
    new_arr[:place] = np.zeros((new_arr[:place].shape))
    return new_arr

shift_array(a,2)

# array([[ 0,  0,  0],
#        [ 0,  0,  0],
#        [ 1,  2,  3],
#        [ 4,  5, 6]])
1 голос
/ 21 июня 2019

Если вы хотите сделать это вручную, вот как работает нарезка:

import numpy as np

A = np.array([[ 1, 2, 3],
              [ 4, 5, 6],
              [ 7, 8, 9],
              [10, 11, 12]])

A[1:,:] = A[:3,:]
A[:1,:] = [0, 0, 0]
...