用Python的numpy包

Question

创建一个一组数组,其内容为[1,2,3,4,5,6,7,8,9,10]。在此基础上生成一个新的数组,要求其内容为[1,2,3,4],[2,3,4,5],[3,4,5,6],…,[7,8,9,10]

Answer 1

import numpy as np

a=np.array([1,2,3,4,5,6,7,8,9,10])
b=np.array([[None]*4 for i in range(len(a)-3)])
for i in range(len(a)-3):
    b[i]=a[i:i+4]
print(b)