Sorting a 2D numpy array by multiple axes

by

Using lexsort:

import numpy as np    
a = np.array([(3, 2), (6, 2), (3, 6), (3, 4), (5, 3)])

ind = np.lexsort((a[:,1],a[:,0]))    

a[ind]
# array([[3, 2],
#       [3, 4],
#       [3, 6],
#       [5, 3],
#       [6, 2]])

a.ravel() returns a view if a is C_CONTIGUOUS. If that is true,
@ars’s method, slightly modifed by using ravel instead of flatten, yields a nice way to sort a in-place:

a = np.array([(3, 2), (6, 2), (3, 6), (3, 4), (5, 3)])
dt = [('col1', a.dtype),('col2', a.dtype)]
assert a.flags['C_CONTIGUOUS']
b = a.ravel().view(dt)
b.sort(order=['col1','col2'])

Since b is a view of a, sorting b sorts a as well:

print(a)
# [[3 2]
#  [3 4]
#  [3 6]
#  [5 3]
#  [6 2]]

More Related Contents:

  1. Sorting arrays in NumPy by column
  2. Rank items in an array using Python/NumPy, without sorting array twice
  3. A fast way to find the largest N elements in an numpy array
  4. How can I “zip sort” parallel numpy arrays?
  5. Find elements of array one nearest to elements of array two
  6. Sort invariant for numpy.argsort with multiple dimensions
  7. Efficiently sorting a numpy array in descending order?
  8. sort eigenvalues and associated eigenvectors after using numpy.linalg.eig in python
  9. Sorting a python array/recarray by column
  10. argsort for a multidimensional ndarray
  11. Sort array’s rows by another array in Python
  12. ‘and’ (boolean) vs ‘&’ (bitwise) – Why difference in behavior with lists vs numpy arrays?
  13. Use a.any() or a.all()
  14. Restart cumsum and get index if cumsum more than value
  15. ValueError: numpy.dtype has the wrong size, try recompiling
  16. Implement MATLAB’s im2col ‘sliding’ in Python
  17. Sorting Python list based on the length of the string
  18. subsampling every nth entry in a numpy array
  19. How to plot empirical cdf (ecdf)
  20. Matplotlib: TypeError: can’t multiply sequence by non-int of type ‘numpy.float64’
  21. Are numpy’s basic operations vectorized, i.e. do they use SIMD operations?
  22. Simple wrapping of C code with cython
  23. pandas distinction between str and object types
  24. numpy reverse multidimensional array
  25. What algorithm does Python’s built-in sort() method use?
  26. Form a big 2d array from multiple smaller 2d arrays
  27. Load CSV to Pandas MultiIndex DataFrame
  28. Python pandas – new column’s value if the item is in the list
  29. Digitizing an analog signal
  30. calculate distance of 2 list of points in numpy

Leave a Comment