首页 > 解决方案 > 在列表理解中使用就地列表方法的替代方法?

问题描述

我知道就地列表方法返回None而不是变异列表。据我所知,这使得将这些方法用作列表推导的内部逻辑的一部分是不可能的。

创建一个列表理解的最pythonic方法是什么,它的成员是由改变其他列表产生的?换句话说:这个(非工作)行的最佳替代方案是什么:

new_list = [old_list.insert(0, "X") for old_list in list_of_old_lists]

这会导致Nones 列表,因为list.insert()返回None.

在没有大量切片和连接的情况下,是否根本不可能在优雅的单行代码中做到这一点?

为了说明我的问题,上面的例子是微不足道的,但实际上我想在更复杂的情况下这样做,而不是多个嵌套的“for”循环。

这是我正在尝试做的简化示例:

word = 'abcdefg'

variations_list = []
characters_to_insert = ['X', 'Y', 'Z']

for character in characters_to_insert:
    for position in range(len(word) + 1):
        w = list(word)
        w.insert(position, character)
        this_variation = ''.join(w)
        variations_list.append(this_variation)

for v in variations_list:
    print(v)

这可以很好地使用嵌套的“for”循环,就像这样(我的实际应用程序比这个示例更复杂/冗长)。

但我不能使用列表推导做同样的事情,因为“插入”方法返回None

variations_list_comprehension = [list(word).insert(position, character) for position in range(len(word) +1) for character in ['X', 'Y', 'Z']]

for v in variations_list_comprehension:
    print(v)

结果是None值列表,因为就地突变返回“无”。

标签: pythonstringpython-3.xlistlist-comprehension

解决方案


并非所有事情都应该或需要通过理解来解决。for-loops 还不错 - 有时它们比理解更好,尤其是因为人们倾向于避免自动在一行中做太多事情。

但是,如果您真的想要一个列表理解解决方案,我会使用一个包装就地操作的辅助函数:

def insert(it, index, value):
    lst = list(it)
    lst.insert(index, value)
    return lst

[''.join(insert(word, position, character)) for position in range(len(word) +1) for character in ['X', 'Y', 'Z']]

但是,要真正等同于您的循环解决方案,您需要在理解中交换循环:

[''.join(insert(word, position, character)) for character in ['X', 'Y', 'Z'] for position in range(len(word) +1)]

这里的优点是包装就地方法可以在很多情况下应用,它不仅仅在这种情况下工作(你可以用这种方式包装任何就地函数)。它很冗长,但它的可读性和可重用性非常好!


我个人会使用带有循环的生成器函数,因为您可以使用它来创建列表,但您也可以按需创建项目(不需要列表):

def insert_characters_everywhere(word, chars):
    for character in characters_to_insert:
        for position in range(len(word) + 1):
            lst = list(word)
            lst.insert(position, character)
            yield ''.join(lst)

list(insert_characters_everywhere('abcdefg', ['X', 'Y', 'Z']))

推荐阅读