python 列表元素明智的条件增量

Question

我已经搜索了一段时间，基本上我是在尝试有条件地按元素递增另一个列表的元素列表...

我的代码如下，但是有更好的方法吗？列表理解， map ??

我认为像 ~+= 这样的元素运算符来自 http://www.python.org/dev/peps/pep-0225/真的很好，但为什么要推迟？

for i in range(1,len(s)):
        if s[i]<s[0]:
            s[i]+=p[i]

根据你们的一些好的反馈，我重新编码如下

i=s<s[0]
s[i]+=p[i]

和s,p都是数组。

p.s 仍然比 matlab 慢 5 倍我的代码之一。

Answer 1

这是一个快速版本:

# sample data
s = [10, 5, 20]
p = [2,2,2]

# As a one-liner.  (You could factor out the lambda)
s = map(lambda (si, pi): si + pi if si < s[0] else si, zip(s,p))

# s is now [10, 7, 20]

这假设len(s) <= len(p)

希望这对您有所帮助。让我知道。 祝你好运。 :-)