将带有赋值的循环转换为理解

Question

将循环转换为理解很简单:

mylist = []
for word in ['Hello', 'world']:
    mylist.append(word.split('l')[0])

至

mylist = [word.split('l')[0] for word in ['Hello', 'world']]

但是当循环涉及为引用分配值时,我不确定如何继续.

mylist = []
for word in ['Hello', 'world']:
    split_word = word.split('l')
    mylist.append(split_word[0]+split_word[1])

理解最终看起来像这样:

mylist = [word.split('l')[0]+word.split('l')[1] for word in ['Hello', 'world']]

这计算word.split('l')多次,而循环仅计算一次并保存参考.我尝试过以下方法:

mylist = [split_word[0]+split_word[1] for word in ['Hello', 'world'] with word.split('l') as split_word]

失败,因为with不这样做,并且:

mylist = [split_word[0]+split_word[1] for word in ['Hello', 'world'] for split_word = word.split('l')]

这也不起作用.我知道通过拆包*,**但我不确定这里适合的地方.是否有可能将这些类型的循环变成理解,希望以一种整洁的方式？

Answer 1

您无法直接将该循环转换为理解.作为表达式的理解只能包含表达式,而赋值是语句.

但是,这并不意味着没有选择.

首先,以split两次调用为代价,您可以这样做:

mylist = [word.split('l')[0]+word.split('l')[1] for word in ['Hello', 'world']]

但你不想打split两次电话.

最常见的方法是使用一系列生成器表达式(最后一个列表理解)来转换事物:

words = (word.split('l') for word in ['Hello', 'world'])
mylist = [w[0]+w[1] for w in words]

如果您真的想将所有内容合并到一个表达式中,您可以:

mylist = [w[0]+w[1] for w in 
          (word.split('l') for word in ['Hello', 'world'])]

但除非你真的需要它在一个表达式中,否则它可能更具可读性.

在这种情况下,更具体的方法是w[0]+w[1]用不需要引用w两次的等价物替换:

mylist = [''.join(word.split('l')[:2]) for word in ['Hello', 'world']]

你也可以随时概括这一点.您可以将任何表达式转换为函数,这意味着您可以通过将其作为参数传递给该函数来避免评估它的任何部分.如果没有能够执行您想要的功能,请将其写入:

def join_up(split_word):
    return split_word[0]+split_word[1]
mylist = [join_up(word.split('l')) for word in ['Hello', 'world']]

如果您需要将所有内容组合成一个表达式而不重复任何工作,那么它可能不是很漂亮:

mylist = [(lambda split_word: split_word[0]+split_word[1])(word.split('l')) 
          for word in ['Hello', 'world']]

但最终,除非我已经有了一个能够满足我需要的功能,否则我会使用生成器链表达式解决方案.

或者,当然,只需将其保持在显式循环中; for循环没有任何问题,如果中间临时变量使代码更清晰,那么没有比使用赋值语句更好的方法.

Answer 2

要有创意.

mylist = [''.join(word.split('l')[:2]) for word in ['Hello', 'world']]

...

mylist = [''.join(operator.itemgetter(0, 1)(word.split('l')))
            for word in ['Hello', 'world']]