考虑下面的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]])
我该如何实现?