python 旋转列表

╰半夏微凉° 2022-09-12 09:54 184阅读 0赞

对列表进行旋转，

比如[1,2,3,4,5]

旋转1为[2,3,4,5,1]

旋转2为[3,4,5,1,2]

同时支持list列表和numpy矩阵。

旋转一维的：

  1. from collections import deque
  3. import numpy as np
  6. data = np.array([1, 2, 3])
  7. items = deque(np.array(data))
  11. items.rotate(0)
  13. print(items)

旋转二维的：

  1. from collections import deque
  3. import numpy as np
  5. data=[[1,2],[2,5],[3,6],[4,5]]
  7. items = deque(np.array(data))
  11. items.rotate(0)
  13. print(items)

结果可以转回list：

  1. from collections import deque
  3. import numpy as np
  5. data=[[1,2],[2,5],[3,6],[4,5]]
  7. # data = np.array([1, 2, 3])
  8. items = deque(np.array(data))
  12. items.rotate(0)
  14. print(list(items))

结果可以转回numpy：

  1. from collections import deque
  3. import numpy as np
  5. data=[[1,2],[2,5],[3,6],[4,5]]
  7. # data = np.array([1, 2, 3])
  8. items = deque(np.array(data))
  12. items.rotate(0)
  14. print(np.array(items))

参考：

在python中旋转列表的有效方法-CSDN论坛

发表评论

表情：
评论列表 （有 0 条评论，184人围观）

还没有评论，来说两句吧...

相关阅读