如何高效截断嵌套列表元素？单次遍历实现方法求教

我有一个较长的嵌套列表（list of lists），其中每个子列表的第一个元素是长字符串，需要将其截断为较短字符串。目前我需遍历列表两次：第一次截断第一个元素，第二次格式化输出，代码如下：

for i in range(len(prefData)):
    temp = prefData[i][0]
    temp = temp[:40]
    prefData[i][0] = temp
for row in prefData:
    print('formatting strings'.format(*row))

请问能否仅遍历一次列表，在遍历过程中同时完成截断操作？我希望原地修改列表元素，而非仅在输出时进行截断。

当然可以！你完全可以把截断操作和输出逻辑合并到同一次遍历里，既完成原地修改，又不用多跑一遍循环。

直接遍历列表的每一行（同时获取索引或者直接修改元素）就行，这里有两种简洁的写法：

写法一：通过索引遍历（和你原来的思路衔接更顺） #

for i in range(len(prefData)):
    # 先截断并原地修改第一个元素
    prefData[i][0] = prefData[i][0][:40]
    # 立刻格式化输出当前行
    print('formatting strings'.format(*prefData[i]))

写法二：直接遍历元素（更Pythonic） #

如果你不想用索引，也可以通过枚举（enumerate）来同时拿到索引和元素，这样代码可读性更好：

for idx, row in enumerate(prefData):
    # 截断第一个元素并原地修改
    row[0] = row[0][:40]
    # 格式化输出
    print('formatting strings'.format(*row))

这两种写法都只遍历一次列表，在处理每一行的时候先完成截断并原地更新列表元素，紧接着就执行输出操作，完美满足你的需求。而且第二种用enumerate的写法是Python里处理这类遍历修改场景的惯用方式，代码更清晰直观。

内容的提问来源于stack exchange，提问作者Patrick