Question

在原位反转数组的部分的最佳（最快/最pythonic）方法是什么？

例如，

def reverse_loop(l,a,b):
    while a < b:
        l[a],l[b] = l[b],l[a]
        a += 1
        b -= 1

现在以后

l = list(range(10))
reverse_loop(l,2,6)

l是[0, 1, 6, 5, 4, 3, 2, 7, 8, 9]。

A，Python中的循环效率低下，因此需要一种更好的方法，e.g.，

def reverse_slice(l,a,b):
    l[a:b+1] = l[b:a-1:-1]

和reverse_slice(l,2,6)将l恢复为其原始值。

A，这不适用于边界案件：reverse_slice(l,0,6) 将l截断为[7, 8, 9]，因为l[a:-1:-1]应该是l[a::-1]。

那么，正确的方法是什么？

Answer 1

怎么样？

def reverse_slice(l, a, b):
    l[a:b] = l[a:b][::-1]

l = list(range(10))
reverse_slice(l, 0, 6)  # excludes l[6]
print(l)

输出：

[5, 4, 3, 2, 1, 0, 6, 7, 8, 9]

内置函数reversed的替代方案：

def reverse_func(l, a, b):
    l[a:b] = reversed(l[a:b])

在我的测试中，切片速度比使用reversed快1.2倍至1.5倍。

Answer 2

[6::-1]可以写为[6:None:-1]：

def reverse_slice(l,a,b):
    a1 = None if a==0 else a-1
    l[a:b+1] = l[b:a1:-1]

In [164]: y=x.copy(); reverse_slice(y,1,6);y                                    
Out[164]: [0, 6, 5, 4, 3, 2, 1, 7, 8, 9]
In [165]: y=x.copy(); reverse_slice(y,0,6);y                                    
Out[165]: [6, 5, 4, 3, 2, 1, 0, 7, 8, 9]

就地反转数组的一部分

2 个答案: