是否可以一次将地图功能应用于多个值?
像这样的东西
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]})
答案 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])