蟒蛇列表元素明智的条件增量

Question

我一直在寻找这一段时间，基本上我试图通过另一份清单，逐元素的有条件增加元素的列表...

我的代码以下，但有更好的方法来做到这一点？列表理解，地图??

我觉得像〜+ = 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] 

和一些好反馈，p均为阵列。

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

Answer 1

如果你不希望创建一个新的数组，那么你的选择是：

1. 你提出什么（尽管你可能要取决于Python版本使用的xrange ）
2. 对s和p使用numpy数组。然后，如果s和p的长度相同，则可以执行类似s[s<s[0]] += p[s<s[0]]的操作。
3. 使用Cython加快你的建议。

Answer 2

入住这太问题：

• Merging/adding lists in Python

基本上是这样的：

[sum(a) for a in zip(*[s, p]) if a[0] < 0] 

例子：

>>> [sum(a) for a in zip(*[[1, 2, 3], [10, 20, 30]]) if a[0] > 2] 
[33] 

要CLAR ify，这是zip的功能：

>>> zip(*[[1, 2, 3], [4, 5, 6]]) 
[(1, 4), (2, 5), (3, 6)] 

它将两个（或多个）列表连接到一个元组列表中。您可以测试每个元组元素的条件。

Answer 3

下面是一个简单的版本：

# 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)

希望这有助于。让我知道。 祝你好运。 :-)

Answer 4

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