如何解决从中间对列表进行排序的最 Pythonic 方式?
我设法编写了以下代码,但仅适用于奇数大小的列表。为此编写一个偶数版本并不难,但我觉得必须有更通用和“Pythonic”的方式来编码它。我真的很惊讶这个微不足道的问题以前从未被问过。
# Odd length,works
lst = [1,3,5,7,9,10,8,6,4,2,0] # Expected output [10,1,0]
# Even length,doesn't work
#lst = [1,0] # Expected output [9,0]
mid = len(lst) // 2
dist = 1
res = [lst[mid]]
while dist <= mid:
res += sorted([lst[mid - dist],lst[mid + dist]],reverse=True)
dist += 1
assert len(res) == len(lst)
print(res)
注意第一个数字后面的序列必须遵循距离+-1、+-2、+-3、...
例如,[5,1]
的预期输出是 [3,1]
; [5,4]
的预期输出是 [1,4]
另外,如果有人可以提供带有 reverse=True
或 False
选项的函数,这样用户可以决定是先选择更大的数字还是更小的数字,那就太好了。
解决方法
尝试以下操作(虽然我更喜欢下面的递归版本):
lst_odd = [1,3,5,7,9,10,8,6,4,2,0]
lst_even = [1,0]
def sorted_from_middle(lst,reverse=False):
left = lst[len(lst)//2-1::-1]
right = lst[len(lst)//2:]
output = [right.pop(0)] if len(lst) % 2 else []
for t in zip(left,right):
output += sorted(t,reverse=reverse)
return output
print(sorted_from_middle(lst_odd)) # [10,1]
print(sorted_from_middle(lst_even)) # [8,1]
print(sorted_from_middle(lst_odd,True)) # [10,1,0]
print(sorted_from_middle(lst_even,True)) # [9,0]
print(sorted_from_middle([5,1],True)) # [3,1]
print(sorted_from_middle([5,4],True)) # [1,4]
想法是将输入列表分成两部分,left
和right
;通过首先将最中心的项目(如果有)附加到输出列表(在代码中这是通过从 right
窃取一个项目来完成的)来使它们具有相等的长度。让 left
反转,以便在 left
和 right
中,第一个元素最靠近中心。然后使用 zip()
同时取两个离中心距离为 1 的元素,比较它们,并相应地将它们附加到输出列表中。然后移至下一对项目,依此类推。
顺便说一下,您也可以使用递归函数来做到这一点,在我看来,它可以更干净地处理偶数/奇数情况。
def sorted_from_middle(lst,reverse=False):
if len(lst) <= 1:
return lst
tail = sorted([lst[-1],lst[0]],reverse=reverse)
return sorted_from_middle(lst[1:-1],reverse) + tail
,
这个想法基本上只是 1) 排除中间元素,如果它是奇数并保存它,然后 2) 对由向后组织的剩余列表前半部分的一个元素组成的每一对进行排序,另一对来自剩余列表的后半部分列表;和 3) 展平对列表,将前一个中间元素添加为第一个元素(如果有)
import numpy
lst1 = [1,0]
lst1_out = [10,0]
lst2 = [1,0]
lst2_out = [9,0]
lst3 = [5,1]
lst3_out = [3,1]
lst4 = [5,4]
lst4_out = [1,4]
def sort_from_middle(lst,reverse=True):
lst = lst.copy()
N = len(lst)
if N == 1:
return lst
elif N == 2:
return sorted(lst,reverse=reverse)
else:
flatten_func = lambda t: [item for sublist in t for item in sublist]
n = int(numpy.floor(N/2.0))
x0 = [ lst.pop(n) ] if ((N%2) == 1) else []
N = len(lst)
return x0 + flatten_func([ sorted([x1,x2],reverse=reverse) for x1,x2 in zip(lst[int(N/2)-1::-1],lst[int(N/2):]) ])
print(lst1)
print(sort_from_middle(lst1))
print(lst1_out)
print(' \n')
print(lst2)
print(sort_from_middle(lst2))
print(lst2_out)
print(' \n')
print(lst3)
print(sort_from_middle(lst3))
print(lst3_out)
print(' \n')
print(lst4)
print(sort_from_middle(lst4))
print(lst4_out)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。