使用map函数进行多次赋值

时间:2013-02-12 05:27:16

标签: python

是否可以一次将地图功能应用于多个值?

像这样的东西

from collections import defaultdict
d['a'] = [1,2,3,4]
d['b'] = [4,5,6,7]
d['a'], d['b'] = map(lambda x,y: (x,y) if x*y % 3 == 0 else (0,0), d['a'], d['b'])
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-129-5191c9205e6f> in <module>()
----> 1 d['a'], d['b'] = map(lambda x,y: (x,y) if x*y % 3 == 0 else (0,0), d['a'], d['b'])

ValueError: too many values to unpack

当然可以单独进行此操作。

 l = map(lambda x,y: x if x*y % 3 == 0 else 0, d['a'],d['b'])
 m = map(lambda x,y: x if x*y % 3 == 0 else 0, d['b'],d['a'])
 d['a'] = l
 d['b'] = m

 d
 defaultdict(<type 'list'>, {'a': [0, 0, 3, 0], 'b': [0, 0, 6, 0]})

4 个答案:

答案 0 :(得分:3)

我想不出让map函数取两个或更多值的方法(我希望有人会在这个帖子中告诉我)。但是你可以借助列表生成器和zip s:

来实现你所需要的
from collections import defaultdict

d = {}
d['a'] = [1,2,3,4]
d['b'] = [4,5,6,7]
d['a'], d['b'] = [list(x) for x in zip( *[(x,y) if x*y % 3 == 0 else (0,0) for (x,y) in zip(d['a'], d['b'])])]

如果您不需要d ['a']和d ['b']作为列表,您可以使最后一行更容易:

d['a'], d['b'] = zip( *[(x,y) if x*y % 3 == 0 else (0,0) for (x,y) in zip(d['a'], d['b'])])

答案 1 :(得分:1)

是的,你可以使用zip来完成。它虽然看起来不是pythonic:

from collections import defaultdict
d = defaultdict(list)
d['a'] = [1,2,3,4]
d['b'] = [4,5,6,7]
d['a'], d['b'] = zip(*map(lambda (x,y): (x,y) if x*y % 3 == 0 else (0,0),
                                                   zip(d['a'], d['b'])))
#out: defaultdict(<type 'list'>, {'a': (0, 0, 3, 0), 'b': (0, 0, 6, 0)})

要获得准确的输出:

d['a'], d['b'] = map(list,zip(*map(lambda (x,y): (x,y) if x*y % 3 == 0 else (0,0), 
                                                           zip(d['a'], d['b']))))
#out: defaultdict(<type 'list'>, {'a': [0, 0, 3, 0], 'b': [0, 0, 6, 0]})

答案 2 :(得分:0)

如果你要做的是这个:

>>> a['b'] = None
>>> a['c'] = None
>>> a['b'], a['c'] = (1,2)
>>> a['b']
1
>>> a['c']
2

是的,但是您可能需要再次查看map函数返回的内容。

答案 3 :(得分:0)

如果您的参数是整数或浮点数,并且您的函数可以使用算术标准运算表达,那么您也可以使用数组来实现此目的

from numpy import array

x_vals = array([1,2,3,4])
y_vals = array([4,5,6,7])
f = lambda x,y: (x*y % 3 == 0)*x

f(x_vals,y_vals)
>>> array([0, 0, 3, 0])