移位数组而不丢失数据python

如果您定义 shift 具体如下:

def shift(a, k):
    return [0 for _ in range(k)] + a

然后你得到

A = [1, 2, 3, 4]
b = shift(A, 2)
print(b) # [0, 0, 1, 2, 3, 4]

如果您想让shift更通用,您可以

def shift(a, k, side=None):
    if shift is None or shift == 'left':
        return [0 for _ in range(k)] + a
    elif shift == 'right':
        return a + [0 for _ in range(k)]

使用它总是更好的 numpy 功能。 np.roll 保留所有数据并执行循环移位:

A = np.array([1, 2, 3, 4])
np.roll(A,2) #roll 2 places
#array([3, 4, 1, 2])

这并不完全是您想要的,但它确实保留了您的所有数据,而且速度很快。

您可以使用 np.pad ,它将比 NP卷 但它更符合您的需求:

np.pad(A,(2,0),mode='constant')
#array([0, 0, 1, 2, 3, 4])

或在两端用0填充

np.pad(A,(2,3),mode='constant')
#array([0, 0, 1, 2, 3, 4, 0, 0, 0])