Pandy: uzyskać pierwsze 10 elementy serii

Question

mam ramki danych w kolumnie tfidf_sorted następująco:

tfidf_sorted 

0 [(morrell, 45.9736796), (football, 25.58352014... 
1 [(melatonin, 48.0010051405), (lewy, 27.5842077... 
2 [(blues, 36.5746634797), (harpdog, 20.58669641... 
3 [(lem, 35.1570832476), (rottensteiner, 30.8800... 
4 [(genka, 51.4667410433), (legendaarne, 30.8800... 

W type(df.tfidf_sorted) powraca pandas.core.series.Series.

Kolumnę utworzony w następujący sposób:

df['tfidf_sorted'] = df['tfidf'].apply(lambda y: sorted(y.items(), key=lambda x: x[1], reverse=True)) 

gdzie tfidf jest słownik.

Jak uzyskać pierwsze 10 par klucz-wartość od tfidf_sorted?

Answer 1

IIUC można użyć:

from itertools import chain 

#flat nested lists 
a = list(chain.from_iterable(df['tfidf_sorted'])) 
#sorting 
a.sort(key=lambda x: x[1], reverse=True) 
#get 10 top 
print (a[:10]) 

Lub jeśli trzeba top 10 w rzędzie dodać [:10]:

df['tfidf_sorted'] = df['tfidf'].apply(lambda y: (sorted(y.items(), key=lambda x: x[1], reverse=True))[:10])