python - 在列表理解中使用就地列表方法的替代方法?
问题描述
我知道就地列表方法返回None
而不是变异列表。据我所知,这使得将这些方法用作列表推导的内部逻辑的一部分是不可能的。
创建一个列表理解的最pythonic方法是什么,它的成员是由改变其他列表产生的?换句话说:这个(非工作)行的最佳替代方案是什么:
new_list = [old_list.insert(0, "X") for old_list in list_of_old_lists]
这会导致None
s 列表,因为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
值列表,因为就地突变返回“无”。
解决方案
并非所有事情都应该或需要通过理解来解决。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']))