2013-02-10 18 views
24

Chcę znaleźć minimalną listę sortowania krotek według danej kolumny. Mam kilka danych ułożonych na przykład jako lista 2-krotna.Pary Tuple, znajdowanie minimum za pomocą pytona

data = [ (1, 7.57), (2, 2.1), (3, 1.2), (4, 2.1), (5, 0.01), 
     (6, 0.5), (7, 0.2), (8, 0.6)] 

Jak mogę znaleźć min zestawu danych przez porównanie drugiej liczby tylko w krotkach?

tj

data[0][1] = 7.57 
data[1][1] = 2.1 

min (dane) = (5, 0.01)

min(data) powraca (1, 7.57), które akceptują jest poprawna dla minimum indeksie 0, ale chcę minimum indeksu 1.

Odpowiedz

44
In [2]: min(data, key = lambda t: t[1]) 
Out[2]: (5, 0.01) 

lub:

In [3]: import operator 

In [4]: min(data, key=operator.itemgetter(1)) 
Out[4]: (5, 0.01) 
1

Mimo że odpowiedź Leva jest prawidłowa, chciałem dodać również metodę sortowania, na wypadek gdyby ktoś był zainteresowany minusem w postaci pierwszych n. Jedną rzeczą do rozważenia jest to, że czas pracy w min Operacja jest O(N) gdzie jest porządek na O(N Log N)

data = [ (1, 7.57), (2, 2.1), (3, 1.2), (4, 2.1), (5, 0.01), (6, 0.5), (7, 0.2), (8, 0.6)] 
data.sort(key=lambda x:x[1]) 
print data 

>>> [(5, 0.01), (7, 0.2), (6, 0.5), (8, 0.6), (3, 1.2), (2, 2.1), (4, 2.1), (1, 7.57)] 

https://www.ics.uci.edu/~pattis/ICS-33/lectures/complexitypython.txt

0

Jeśli jesteś gotów do picia numpy coolaid można używać tych poleceń aby uzyskać krotkę na liście, gdzie pozycja jest minimalna:

Składniki, które to umożliwiają, to zaawansowana funkcja szeregowania numpy i funkcje argsort .

import numpy as np 
#create a python list of tuples and convert it to a numpy ndarray of floats 
data = np.array([ (1, 7.57), (2, 2.1), (3, 1.2), 
        (4, 2.1), (5, 0.01), (6, 0.5), (7, 0.2), (8, 0.6)]) 

print("data is") 
print(data) 

#Generate sortIndices from second column 
sortIndices = np.argsort(data[:,1]) 

print("sortIndices using index 1 is:") 
print(sortIndices) 
print("The column at index 1 is:") 
print(data[:,1]) 
print("Index 1 put into order using column 1") 
print(data[sortIndices,1]) 
print("The tuples put into order using column 1") 
print(data[sortIndices,:]) 
print("The tuple with minimum value at index 1") 
print(data[sortIndices[0],:]) 
print("The tuple with maximum value at index 1") 
print(data[sortIndices[-1],:]) 

która drukuje:

data is 
[[ 1. 7.57] 
[ 2. 2.1 ] 
[ 3. 1.2 ] 
[ 4. 2.1 ] 
[ 5. 0.01] 
[ 6. 0.5 ] 
[ 7. 0.2 ] 
[ 8. 0.6 ]] 

sortIndices using index 1 is: 
[4 6 5 7 2 1 3 0] 

The column at index 1 is: 
[ 7.57 2.1 1.2 2.1 0.01 0.5 0.2 0.6 ] 

Index 1 put into order using column 1 
[ 0.01 0.2 0.5 0.6 1.2 2.1 2.1 7.57] 

The tuples put into order using column 1 
[[ 5. 0.01] 
[ 7. 0.2 ] 
[ 6. 0.5 ] 
[ 8. 0.6 ] 
[ 3. 1.2 ] 
[ 2. 2.1 ] 
[ 4. 2.1 ] 
[ 1. 7.57]] 

The tuple with minimum value at index 1 
[ 5. 0.01] 

The tuple with maximum value at index 1 
[ 1. 7.57] 
Powiązane problemy