如何使用python生成器表达式创建oneliner以多次运行函数并获取列表输出

Question

我想知道是否有一个简单的Pythonic方式(可能使用生成器)在列表中的每个项目上运行一个函数,并产生一个返回列表？

例:

def square_it(x):
    return x*x

x_set = [0,1,2,3,4]
squared_set = square_it(x for x in x_set)

我注意到,当我对此进行逐行调试时,传递给函数的对象是生成器.

因此,我收到一个错误: TypeError: unsupported operand type(s) for *: 'generator' and 'generator'

我知道这个生成器表达式创建了一个传递给函数的生成器,但我想知道是否有一种很酷的方法来完成多次运行该函数只能通过指定一个iterable作为参数？(不修改函数以期望迭代).

在我看来,这种能力对于减少代码行非常有用,因为你不需要创建一个循环来使函数变得有趣,而一个变量将输出保存在列表中.

谢谢!

Answer 1

你想要一个列表理解:

squared_set = [square_it(x) for x in x_set]