Python. Holen Sie Schlüsselpaare (n0, _), (n1, _) mit maximalen Werten für jedes Schlüsselpaar (n0, a), (n0, b) in einem Wörterbuch. Keine triviale Aufgabe

Question

Angenommen, wir haben eine dict wie:

os_stats = { 
    ('USA', 'Mac OS X'): 1, 
    ('Mexico', 'iOS'): 3, 
    ('USA', 'Windows XP'): 2, 
    ('Germany', 'Windows 7'): 9, 
    ('Germany', 'Windows XP'): 7, 
    ('Mexico', 'Windows XP'): 2, 
    ... 
} 

Und ich möchte eine Ausgabe wie:

os_preferences = { 
    ('Mexico', 'iOS'): 3, 
    ('USA', 'Windows XP'): 2, 
    ('Germany', 'Windows 7'): 9, 
     ... 
} 

Wo nur der höchste Wert für jedes Land präsentiert wird. Wie kann ich das erreichen?

Answer 1

Dieses dict Verständnis tut es:

{country:{os:count} for (country,os),count in sorted(os_stats.items(), key=lambda rec:rec[1])} 

Der erste Teil ist dies:

sorted(os_stats.items(), key=lambda rec:rec[1]) 

Das erzeugt:

[(('USA', 'Mac OS X'), 1), 
(('Mexico', 'Windows XP'), 2), 
(('USA', 'Windows XP'), 2), 
(('Mexico', 'iOS'), 3), 
(('Germany', 'Windows XP'), 7), 
(('Germany', 'Windows 7'), 9)] 

Beachten Sie, dass es durch aufsteigende Reihenfolge der Zählung sortiert ist Feld (rec[1]).

Der Rest ist nur Massieren der Daten in eine einzige dict, die den Effekt der Verwerfung der kleineren Werte durch Überschreiben sie, wie es mit den größeren geht.

Answer 2

pandas bekommt den Job in 3 Zeilen getan:

import pandas as pd 
df = pd.DataFrame(os_stats, index=['index']).transpose() 
os_preferences = df[df['index'] == df.groupby(level=[0])['index'].transform(max)].to_dict()['index'] 

# output: 
# {('Mexico', 'iOS'): 3, 
# ('USA', 'Windows XP'): 2, 
# ('Germany', 'Windows 7'): 9}