将函数映射到变量列表并覆盖

时间:2014-10-27 19:09:17

标签: python

想象一下,我有一个简单的函数,它将返回一个修改后的值:

def foo(x): 
   return x**2

我的代码中有4个变量A - D,我想在其上运行。我希望它能够覆盖A - D.现在,我正在手动执行此操作:

A = foo(A)
B = foo(B) 
etc...

我知道我可以做列表理解来做到这一点,但是我需要从列表中重新分配每个变量。像:

new = [foo(x) for x in [A, B, C, D]]
A,B,C,D = new[0], new[1], new[2], new[3]

我想我正在寻找一些地图函数,它会覆盖变量名称,但不确定哪个!

由于

1 个答案:

答案 0 :(得分:2)

你可以这样做:

A,B,C,D = [foo(x) for x in [A, B, C, D]]