python - 如何将网格中的所有字母向左移动？

Question

所以我在一些有用的 stackoverflow 用户的帮助下编写了一个生成网格的代码，这就是......

import random
import string

for x in range(4):
    for y in range(4):
        print( '['+random.choice(string.ascii_letters)+']',end='' )
    print()

现在我想创建一个函数 moveleft(grid) 它将每列向左移动

例如，如果我的网格一开始看起来像这样..

[k][b][s][v]
[u][w][p][O]
[t][i][i][K]
[n][J][r][f]

然后执行moveleft(grid)后，这将是我的网格，

[v][k][b][s]
[O][u][w][p]
[K][t][i][i]
[f][n][J][r]

我应该写什么才能让 moveleft(grid) 发挥上面提供的示例的作用？

Answer 1

numpy.roll做这个。例如，

>>> import numpy as np
>>> grid = np.array([[1,2,3],[4,5,6],[7,8,9]])
>>> grid
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])
>>> np.roll(grid, shift=1, axis=1)
array([[3, 1, 2],
       [6, 4, 5],
       [9, 7, 8]])