Comment diviser une liste en fonction d'une condition ?

Question

Quelle est la meilleure façon, à la fois esthétiquement et du point de vue des performances, de diviser une liste d'éléments en plusieurs listes sur la base d'une condition ? L'équivalent de :

good = [x for x in mylist if x in goodvals]
bad  = [x for x in mylist if x not in goodvals]

Existe-t-il une manière plus élégante de procéder ?

Voici le cas d'utilisation réel, pour mieux expliquer ce que j'essaie de faire :

# files looks like: [ ('file1.jpg', 33L, '.jpg'), ('file2.avi', 999L, '.avi'), ... ]
IMAGE_TYPES = ('.jpg','.jpeg','.gif','.bmp','.png')
images = [f for f in files if f[2].lower() in IMAGE_TYPES]
anims  = [f for f in files if f[2].lower() not in IMAGE_TYPES]

Answer 1

Je me suis tourné vers numpy pour résoudre ce problème en limitant les lignes et en le transformant en une fonction simple.

J'ai pu obtenir une mesure conditionnelle, séparant une liste en deux, à l'aide de np.where pour séparer une liste. Cela fonctionne pour le numérique, mais on pourrait l'étendre aux chaînes de caractères et aux listes, je crois.

Le voici...

from numpy import where as wh, array as arr

midz = lambda a, mid: (a[wh(a > mid)], a[wh((a =< mid))])
p_ = arr([i for i in [75, 50, 403, 453, 0, 25, 428] if i])
high,low = midz(p_, p_.mean())

Answer 2

Si votre souci est de ne pas utiliser deux lignes de code pour une opération dont la sémantique n'a besoin que d'une seule fois, il vous suffit d'envelopper certaines des approches ci-dessus (même la vôtre) dans une seule fonction :

def part_with_predicate(l, pred):
    return [i for i in l if pred(i)], [i for i in l if not pred(i)]

Il ne s'agit pas d'une approche lazy-eval et elle itère deux fois dans la liste, mais elle vous permet de partitionner la liste en une seule ligne de code.

Answer 3

def partition(pred, seq):
  return reduce( lambda (yes, no), x: (yes+[x], no) if pred(x) else (yes, no+[x]), seq, ([], []) )