使用 numpy,您可以使用这些命令来获取列表中项最小的元组:
使这项工作的成分是 numpy 的高级数组切片和 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],:])
哪个打印:
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]