Question

在Python 3中，是否有一种方法可以用lambda函数（也许用groupList来替换map()函数中的for循环。

def groupList(input_list, output_list=[]):
    for i, (v, w) in enumerate(zip(input_list[:-2], input_list[2:])):
        output_list.append(f'{input_list[i]} {input_list[i+1]} {input_list[i+2]}')
    return output_list

print(groupList(['A', 'B', 'C', 'D', 'E', 'F', 'G']))

（groupList函数的输出为['A B C', 'B C D', 'C D E', 'D E F', 'E F G']）

Answer 1

解决方案1：

def groupList(input_list):
    return [' '.join(input_list[i:i+3]) for i in range(len(input_list) - 2)]

解决方案2：

def groupList(input_list):
    return list(map(' '.join, (input_list[i:i+3] for i in range(len(input_list) - 2))))

Answer 2

除了先前的解决方案之外，一种更有效（但不太简洁）的解决方案是先计算完整的串联，然后对其进行切片。

from itertools import accumulate

def groupList(input_list):
    full_concat = ' '.join(input_list)
    idx = [0]
    idx.extend(accumulate(len(s) + 1 for s in input_list))
    return [full_concat[idx[i]:idx[i+3]-1] for i in range(len(idx) - 3)]

如何枚举/压缩为lambda

2 个答案: