élément de la liste python incrément sage conditionnel

Question

J'ai cherché pendant un certain temps, au fond, je suis en train d'incrémenter conditionnellement une liste d'éléments par une autre liste, élément par élément ...

mon code suit, mais est il y a une meilleure façon de le faire? liste de compréhension, carte ??

Je pense qu'un opérateur par élément comme ~ + = de http://www.python.org/dev/peps/pep-0225/ serait vraiment bon, mais pourquoi est-il différé?

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

basé sur quelques bons retours de vous les gars je recodé à ce qui suit

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

et s, p sont les deux tableaux.

p.s encore plus lent que matlab 5 fois pour l'un de mes codes.

Answer 1

Si vous ne voulez pas créer un nouveau tableau, vos options sont:

1. Qu'est-ce que vous proposez (bien que vous pouvez utiliser xrange selon la version python
2. Utilisez des tableaux Numpy pour s et p. Ensuite, vous pouvez faire quelque chose comme s[s<s[0]] += p[s<s[0]] si s et p ont la même longueur.
3. Utilisez Cython pour accélérer ce que vous avez proposé.

Answer 2

Cochez cette question SO:

• Merging/adding lists in Python

En gros, quelque chose comme:

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

Exemple:

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

Pour clar ifier, voici ce que zip ne:

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

Il Assemble deux (ou plus) des listes dans une liste de tuples. Vous pouvez tester les conditions sur les éléments de chacun des tuples.

Answer 3

Voici une version rapide:

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

Cela suppose que len(s) <= len(p)

Hope this helps. Faites-moi savoir. Bonne chance. :-)

Answer 4

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