python - 如何按列对二维列表进行排序

Question

如何摆脱这些 if 语句并通过使用 for 循环而不使用 numpy 使其更简单

 def sortColumns(grid):
    if grid[0][0] > grid[1][0]:
        grid[0][0],grid[1][0]= grid[1][0],grid[0][0]
    if grid[2][0] < grid[1][0]:
        grid[2][0],grid[1][0]= grid[1][0],grid[2][0]
    if grid[1][0] < grid[0][0]:
        grid[1][0],grid[0][0] = grid[0][0],grid[1][0]

    if grid[0][1] > grid[1][1]:
        grid[0][1],grid[1][1]= grid[1][1],grid[0][1]
    if grid[2][1] < grid[1][1]:
        grid[2][1],grid[1][1]= grid[1][1],grid[2][1]
    if grid[1][1] < grid[0][1]:
        grid[1][1],grid[0][1] = grid[0][1],grid[1][1]

    if grid[0][2] > grid[1][2]:
        grid[0][2],grid[1][2]= grid[1][2],grid[0][2]
    if grid[2][2] < grid[1][2]:
        grid[2][2],grid[1][2]= grid[1][2],grid[2][2]
    if grid[1][2] < grid[0][2]:
        grid[1][2],grid[0][2] = grid[0][2],grid[1][2]

for x in grid:
    for y in x:
        print(str(y).center(3,' '),end= ' ')
    print()

示例输出需要与此类似:

Enter the number of rows: 3
Enter the number of columns: 3
   -3  58 -46 
   29  40 -65 
   26 -53 -55 
Sorted list by column is 

  -3 -53 -65 
  26  40 -55 
  29  58 -46 

Answer 1

使用 zip 并解压 * ，您可以转置，排序每个子列表，然后再次转置:

>>> lst = [[-3, 58, -46], [29, 40, -65], [26, -53, -55]]
>>> result = zip(*map(sorted, zip(*lst)))
>>> result
[(-3, -53, -65), (26, 40, -55), (29, 58, -46)]

在 Python 3.x 中，result将是 zip目的;您需要调用 list就在上面。

如果您想将结果保留为列表列表(而不是元组)，您可以使用列表理解来实现:

>>> [list(s) for s in zip(*map(sorted, zip(*lst)))]
[[-3, -53, -65], [26, 40, -55], [29, 58, -46]]