import numpy as np #data = np.sin(np.arange(20)).reshape(5,4) #print data #ind = data.argmax(axis=0) #print ind #data_max = data[ind, xrange(data.shape[1])] #print data_max all(data_max == data.max(axis=0))True
a = np.arange(0, 40, 10) b = np.tile(a, (3, 5)) print b[[ 0 10 20 30 0 10 20 30 0 10 20 30 0 10 20 30 0 10 20 30] [ 0 10 20 30 0 10 20 30 0 10 20 30 0 10 20 30 0 10 20 30] [ 0 10 20 30 0 10 20 30 0 10 20 30 0 10 20 30 0 10 20 30]]
a = np.array([[4, 3, 5], [1, 2, 1]]) #print a #b = np.sort(a, axis=1) #print b #b #a.sort(axis=1) #print a a = np.array([4, 3, 1, 2]) j = np.argsort(a) print j print a[j][2 3 1 0] [1 2 3 4]
236

被折叠的 条评论
为什么被折叠?



