Réduire de clés en python

Je suis en train de réfléchir à la manière la plus efficace de le faire en python.

Supposons que j'ai une liste de tuples:

[('dog',12,2), ('cat',15,1), ('dog',11,1), ('cat',15,2), ('dog',10,3), ('cat',16,3)]

Et supposons que j'ai une fonction qui prend deux de ces n-uplets et les combine:

def my_reduce(obj1, obj2):
    return (obj1[0],max(obj1[1],obj2[1]),min(obj1[2],obj2[2]))

Comment puis-je effectuer un efficace, réduire les par la "clef" où est la clé ici pourrait être la première valeur, de sorte que le résultat final serait quelque chose comme:

[('dog',12,1), ('cat',16,1)]

vouliez-vous dire min(obj1[2],obj2[2])
bonne prise, merci! Je l'ai fixée ci-dessus
cela ressemble à quelque chose de bien adapté pour les pandas
Hey @mgoldwasser, 2 ans de retard, mais voici une autre façon: stackoverflow.com/a/48343896/5858851. BTW, je suis l'ancien collègue de travail, vous pensez que je suis.

InformationsquelleAutor mgoldwasser | 2015-04-29

python reduce

5

Si vous souhaitez utiliser votre my_reduce et reduce, vous pouvez le faire de cette façon. Il est assez court, en fait:

Préparation:
```
from itertools import groupby
from operator import itemgetter

pets = [('dog',12,2), ('cat',15,1), ('dog',11,1), ('cat',15,2), ('dog',10,3), ('cat',16,3)]

def my_reduce(obj1, obj2):
    return (obj1[0],max(obj1[1],obj2[1]),min(obj1[2],obj2[2]))
```
Solution:
```
print [reduce(my_reduce, group)
       for _, group in groupby(sorted(pets), key=itemgetter(0))]
```
De sortie:
```
[('cat', 16, 1), ('dog', 12, 1)]
```
- Pouvez-vous me dire quelle est ta syntaxe/abréviation que vous avez utilisé à l'intérieur de l'instruction print? Il apparaît un appel de fonction, suivi par boucle for et une variable définie par la boucle for est passé dans l'appel de fonction. réduire(my_reduce, groupe) pour _, groupe dans groupby(triés(animaux domestiques), key=itemgetter(0))
- C'est une "compréhension de liste".
InformationsquelleAutor Stefan Pochmann

Sinon, si vous avez pandas installé:

import pandas as pd

l = [('dog',12,2), ('cat',15,1), ('dog',11,1), ('cat',15,2), ('dog',10,3), ('cat',16,3)]

pd.DataFrame(data=l, columns=['animal', 'm', 'n']).groupby('animal').agg({'m':'max', 'n':'min'})
Out[6]: 
         m  n
animal       
cat     16  1
dog     12  1

Pour obtenir le format d'origine:

zip(df.index, *df.values.T) # df is the result above
Out[14]: [('cat', 16, 1), ('dog', 12, 1)]

Je concure 🙂 ... stupide wim et son 0 de la largeur des espaces 😛

InformationsquelleAutor Anzel

Je ne pense pas que reduce est un bon outil pour ce travail, parce que vous devrez d'abord utiliser itertools ou similaires à groupe de la liste par la clé. Sinon, vous serez en comparant cats et dogs et tout l'enfer va barder!

Au lieu d'une simple boucle est fine:

>>> my_list = [('dog',12,2), ('cat',15,1), ('dog',11,1), ('cat',15,2)]
>>> output = {}
>>> for animal, high, low in my_list:
...     try:
...         prev_high, prev_low = output[animal]
...     except KeyError:
...         output[animal] = high, low
...     else:
...         output[animal] = max(prev_high, high), min(prev_low, low)

Alors si vous voulez le format d'origine de retour:

>>> output = [(k,) + v for k, v in output.items()]
>>> output
[('dog', 12, 1), ('cat', 15, 1)]

Note va détruire l'ordre de la liste originale. Si vous souhaitez conserver l'ordre les touches apparaissent en premier dans la, initialiser la sortie avec une OrderedDict à la place.

InformationsquelleAutor wim

si vous voulez vraiment utiliser de réduire je pense que cela fonctionne (il vous donne un dict arrière au lieu d'une liste, mais meh)

def my_reduce(obj1, obj2):
    if not isinstance(obj1,dict):
        return reduce(my_reduce,[{},obj1,obj2])
    try:
        obj1[obj2[0]] = max(obj1[obj2[0]][0],obj2[1]),min(obj1[obj2[0]][1],obj2[2])
    except KeyError:
        obj1[obj2[0]] = obj2[1:]
    return obj1

my_list = [('dog',12,2), ('cat',15,1), ('dog',11,1), ('cat',15,2), ('dog',10,3), ('cat',16,3)]
print reduce(my_reduce,my_list)

Je pense que les deux autres solutions sont mieux cependant

Non, car ce serait l'effondrement de tout ce que sur un seul élément, mais je veux un élément par clé
ok fixe ... en quelque sorte ...

InformationsquelleAutor Joran Beasley

Vous devez vous connecter pour publier un commentaire.