Как получить массив numpy из нескольких списков одинаковой длины и отсортировать вдоль оси?

Question

У меня очень простой вопрос: Как получить массив numpy из нескольких списков одинаковой длины и отсортировать вдоль оси?

Я ищу что-то вроде:

a = [1,1,2,3,4,5,6] 
b = [10,10,11,09,22,20,20] 
c = [100,100,111,090,220,200,200] 
d = np.asarray(a,b,c) 
print d 
>>>[[1,10,100],[1,10,100],[2,11,111].........[6,20,200]] 

2 Вопрос: А если это может быть достигнуто может я сортировать его вдоль оси (например, для по значениям списка б.)?

3-й вопрос: Может ли сортировка выполняться в диапазоне? например. для значений между b + 10 и b-10 при просмотре списка c для дальнейшей сортировки. как

[[1,11,111][1,10,122][1,09,126][1,11,154][1,11,191] 
[1,20,110][1,25,122][1,21,154][1,21,155][1,21,184]] 

Answer 1

Вы можете сжать, чтобы получить массив:

a = [1, 1, 2, 3, 4, 5, 6] 
b = [10, 10, 11, 9, 22, 20, 20] 
c = [100, 100, 111, 90, 220, 200, 200] 
d = np.asarray(zip(a,b,c)) 
print(d) 
[[ 1 10 100] 
[ 1 10 100] 
[ 2 11 111] 
[ 3 9 90] 
[ 4 22 220] 
[ 5 20 200] 
[ 6 20 200]] 

print(d[np.argsort(d[:, 1])]) # a sorted copy 
[[ 3 9 90] 
[ 1 10 100] 
[ 1 10 100] 
[ 2 11 111] 
[ 5 20 200] 
[ 6 20 200] 
[ 4 22 220]] 

Я не знаю, как вы могли бы сделать На месте, вроде, не делая что-то вроде:

d = np.asarray(zip(a,b,c)) 

d.dtype = [("0", int), ("1", int), ("2", int)] 
d.shape = d.size 
d.sort(order="1") 

Ведущий 0 сделал бы 090 восьмеричным в python2 или недействительным синтаксисом в python3, поэтому я удалил его.

Вы также можете отсортировать заархивированные элементы, прежде чем передать:

from operator import itemgetter 
zipped = sorted(zip(a,b,c),key=itemgetter(1)) 


d = np.asarray(zipped) 
print(d) 
[[ 3 9 90] 
[ 1 10 100] 
[ 1 10 100] 
[ 2 11 111] 
[ 5 20 200] 
[ 6 20 200] 
[ 4 22 220]] 

Answer 2

Вы можете использовать np.dstack и np.lexsort. например, если вы хотите, чтобы отсортировать на основе (второй оси) массив b затем a, а затем c:

>>> d=np.dstack((a,b,c))[0] 
>>> indices=np.lexsort((d[:,1],d[:,0],d[:,2])) 
>>> d[indices] 
array([[ 3, 9, 90], 
     [ 1, 10, 100], 
     [ 1, 10, 100], 
     [ 2, 11, 111], 
     [ 5, 20, 200], 
     [ 6, 20, 200], 
     [ 4, 22, 220]])