Longueur de la sortie du générateur

Question

Python fournit une belle méthode pour obtenir la longueur d'un itérable eager, len(x) c'est-à-dire. Mais je n'ai rien trouvé de similaire pour les itérables paresseux représentés par des compréhensions de générateurs et des fonctions. Bien sûr, il n'est pas difficile d'écrire quelque chose comme :

def iterlen(x):
  n = 0
  try:
    while True:
      next(x)
      n += 1
  except StopIteration: pass
  return n

Mais je n'arrive pas à me débarrasser du sentiment que je réimplante un vélo.

(Pendant que je tapais la fonction, une pensée m'a traversé l'esprit : peut-être que cette fonction n'existe pas vraiment, car elle "détruit" son argument. Ce n'est pas un problème pour mon cas, cependant).

P.S. : concernant les premières réponses - oui, quelque chose comme len(list(x)) fonctionnerait aussi, mais cela augmente considérablement l'utilisation de la mémoire.

P.P.S. : revérifié... Ne tenez pas compte du P.S., il semble que j'ai fait une erreur en essayant cela, cela fonctionne bien. Désolé pour le dérangement.

Answer 1

Utilice reduce(fonction, iterable[, initialisateur]) pour une solution purement fonctionnelle et efficace en termes de mémoire :

>>> iter = "This string has 30 characters."
>>> reduce(lambda acc, e: acc + 1, iter, 0)
30

Answer 2

Essayez le more_itertools pour une solution simple. Exemple :

>>> import more_itertools

>>> it = iter("abcde")                                         # sample generator
>>> it
<str_iterator at 0x4ab3630>

>>> more_itertools.ilen(it)
5

Voir ce poste pour un autre exemple appliqué.

Answer 3

D'ailleurs, si vous voulez vérifier si le générateur est vide il suffira de faire quelque chose comme :

def is_empty(generator):
    for item in generator:
        return False
    return True

Answer 4

C'est un hack, mais si vous voulez vraiment avoir len sur un itérable général (en le consommant au passage), vous pouvez créer votre propre version de len .

El len est essentiellement équivalente à la fonction suivante (bien que les implémentations fournissent généralement des optimisations pour éviter la recherche supplémentaire) :

def len(iterable):
    return iterable.__len__()

Par conséquent, nous pouvons définir notre new_len pour essayer cela, et si __len__ n'existe pas, compter nous-mêmes le nombre d'éléments en consommant l'itérable :

def new_len(iterable):
    try:
      return iterable.__len__()
    except AttributeError:
      return sum(1 for _ in iterable)

Ce qui précède fonctionne en Python 2/3, et (pour autant que je sache) devrait couvrir tous les types d'itérables imaginables.