如何通过使用for循环而不使用numpy来摆脱这些if语句并使其变得更简单?
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
发布于 2016-10-30 21:57:33
使用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)]
在Python3.x中,result
将是一个zip
对象;您需要对其调用list
。
如果您希望将结果保存为列表(而不是元组)的列表,则可以使用列表理解来实现:
>>> [list(s) for s in zip(*map(sorted, zip(*lst)))]
[[-3, -53, -65], [26, 40, -55], [29, 58, -46]]
https://stackoverflow.com/questions/40334002
复制相似问题