python - 如何在 python 中使用列表理解修改列表列表中的特定项目

Question

我只是想知道是否有更好的方法使用列表理解来修改列表列表中的特定项目？在下面的示例中，我对列表中每个列表的第二项进行平方，但是，我不想消除内部列表理解。

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

nl = [[num**2 if i==1 else num for i, num in enumerate(x)] for x in l]
print nl

Answer 1

不确定如何保持内心的理解，但你可以这样做:

def square_idx_one(sl):
    sl[1] **= 2
    return sl

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

nl = [square_idx_one(sl) for sl in l]
print (nl)

结果:

[[1, 4, 3], [4, 25, 6], [7, 64, 9]]

但如果你正在修改原始版本，我认为 for 循环可能会在性能方面占据优势，更不用说内存了