python修改函数中的列表切片

Question

考虑以下代码：

def func1(a):
    a[:] = [x**2 for x in a]

a = range(10)
print a  #prints [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
func1(a[:5])
print a  #also prints [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

我希望发送列表的一部分a并在函数内更改它。我的预期输出是

[0, 1, 4, 9, 16, 5, 6, 7, 8, 9]

哪种方式是这样做的惯用方式？

谢谢！

Answer 1

如果对列表进行切片，则只会修改一个副本，因此您想要执行的操作不会以您想要的形式运行。

但是您可以将一个可选slice对象传递给func1，如果不是None，则使用它来执行切片分配（否则使用[:]）

我会执行以下操作（使用 alambda来避免复制/粘贴公式和使用生成器表达式来避免创建无用的临时列表：

def func1(a,the_slice=None):
    e = lambda y : (x**2 for x in y)
    if the_slice:
        a[the_slice] = e(a[the_slice])
    else:
        a[:] = e(a)

测试：

a = list(range(10))
func1(a)
print(a)
a = list(range(10))
func1(a,slice(5))   # stop at 5
print(a)
a = list(range(10))
func1(a,slice(5,len(a),2))  # start at 5 / stop at the end, stride/step 2
print(a)

结果：

[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
[0, 1, 4, 9, 16, 5, 6, 7, 8, 9]
[0, 1, 2, 3, 4, 25, 6, 49, 8, 81]

• 在第一种情况下，列表的总数已更改
• 在第二种情况下，它只改变了前半部分。
• 在第三种情况下，它改变了下半部分，但是 2 中的 1 个值（步幅 = 2）

Answer 2

这将起作用：

a = range(10)

a[:5] = [c**2 for c in a[:5]]