将numpy数组的对角线向左移动并填充

时间:2019-06-08 05:43:20

标签: python arrays numpy shift

考虑下面的2D numpy数组

array1 = np.array([[ 0,  1,  2,  3,  4],
                   [ 5, 21, 15,  8, 12],
                   [ 2,  3,  4,  8,  3],
                   [ 3,  4,  5, 17, 41],
                   [ 4, 20,  7,  5,  6]])

我想要的输出是将数组的对角线向左移动,并用np.nan填充空元素

所需的输出

array([[ 0.,  1.,  2.,  3.,  4.],
       [21., 15.,  8., 12., nan],
       [4.,  8.,  3., nan, nan],
       [17., 41., nan, nan, nan],
       [6., nan, nan, nan, nan]])

我该如何实现?

0 个答案:

没有答案