python - 移动第一个元素以结束 Numpy 数组

Question

我有一个 NumPy 数组，我想将第一个元素移到末尾。我知道如何通过转换为列表并执行插入操作来做到这一点，但我想避免这种情况并以更好和优化的方式进行。

假设这是我的数组，我想将 0 移到末尾:

x = [[ 0 388.13 19.346 412.38 39.594]
    [ 0 388.15 8.5168 416 38.972] 
    [ 0 223.14 156.21 317.91 193.46]]

我知道我可以通过转换为列表以这种方式做到这一点:

X = X.tolist()
for ele in x:
    ele.insert(len(ele), ele.pop(0))

有没有办法不将其转换为列表(使用 numpy)？而且，我正在做的方式效率不高，有什么可能的方法可以使我的代码在速度方面高效，即使是通过将它转换为列表？

预期输出:

x = [[388.13 19.346 412.38 39.594 0]
    [388.15 8.5168 416 38.972 0] 
    [223.14 156.21 317.91 193.46 0]]

Answer 1

尝试 np.roll :np.roll(x, -1, axis=1).tolist()