python - 编辑列表列表中的字符串

Question

我从文本 block 中创建了一个列表列表，其中每个列表包含一行中的所有单词作为单独的元素，如下所示:

listoflists = [['Lorem', 'ipsum', 'dolor', 'sit', 'amet\n']
               ['consectetur', 'adipiscing', 'elit', 'donec', 'iaculis\n']]

我希望能够在列表列表中移动，并从列表列表中每个列表的最后一个元素中删除“\n”。

这是我编写的尝试执行此操作的脚本，但它不起作用，因为当我返回 n 时，它只返回第一个列表的第一个元素。

def remove_linebreaks(input):
for i in data:
    for n in i:
        if '\n' in n:
            n.strip('\n')
            return n
        else:
            return n
return input

还有其他方法可以做到这一点吗？

Answer 1

为什么不直接return n.strip('\n')呢？您可以摆脱 if/else