如何重新转换序列/列表/数组

Question

我有一个数组/列表/熊猫系列：

np.arange(15)
Out[11]: array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14])

我想要的是：

[[0,1,2,3,4,5],
[1,2,3,4,5,6],
[2,3,4,5,6,7],
...
[10,11,12,13,14]]

也就是说，循环地将这些列转换为5列矩阵

原因是我正在为一列温度数据进行特征工程。我想使用最后5个数据作为特征，下一个数据作为目标

最有效的方法是什么？我的数据很大

Answer 1

如果数组的格式如下所示：

arr = np.array([1,2,3,4,5,6,7,8,....])

您可以这样尝试：

recurr_transpose = np.matrix([[arr[i:i+5] for i in range(len(arr)-4)]])