Selecting specific rows and columns from NumPy array

by

As Toan suggests, a simple hack would be to just select the rows first, and then select the columns over that.

>>> a[[0,1,3], :]            # Returns the rows you want
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [12, 13, 14, 15]])
>>> a[[0,1,3], :][:, [0,2]]  # Selects the columns you want as well
array([[ 0,  2],
       [ 4,  6],
       [12, 14]])

[Edit] The built-in method: np.ix_

I recently discovered that numpy gives you an in-built one-liner to doing exactly what @Jaime suggested, but without having to use broadcasting syntax (which suffers from lack of readability). From the docs:

Using ix_ one can quickly construct index arrays that will index the
cross product. a[np.ix_([1,3],[2,5])] returns the array [[a[1,2] a[1,5]], [a[3,2] a[3,5]]].

So you use it like this:

>>> a = np.arange(20).reshape((5,4))
>>> a[np.ix_([0,1,3], [0,2])]
array([[ 0,  2],
       [ 4,  6],
       [12, 14]])

And the way it works is that it takes care of aligning arrays the way Jaime suggested, so that broadcasting happens properly:

>>> np.ix_([0,1,3], [0,2])
(array([[0],
        [1],
        [3]]), array([[0, 2]]))

Also, as MikeC says in a comment, np.ix_ has the advantage of returning a view, which my first (pre-edit) answer did not. This means you can now assign to the indexed array:

>>> a[np.ix_([0,1,3], [0,2])] = -1
>>> a    
array([[-1,  1, -1,  3],
       [-1,  5, -1,  7],
       [ 8,  9, 10, 11],
       [-1, 13, -1, 15],
       [16, 17, 18, 19]])

More Related Contents:

  1. Using numpy to build an array of all combinations of two arrays
  2. Understanding NumPy’s einsum
  3. Intuition and idea behind reshaping 4D array to 2D array in NumPy
  4. Why is numpy’s einsum faster than numpy’s built in functions?
  5. Calculate mean across dimension in a 2D array
  6. Convert a 1D array to a 2D array in numpy
  7. Subsetting a 2D numpy array
  8. Check if two 3D numpy arrays contain overlapping 2D arrays
  9. Selecting Random Windows from Multidimensional Numpy Array Rows
  10. How to split a numpy array in fixed size chunks with and without overlap?
  11. How to find the index of a value in 2d array in Python?
  12. What is the difference between ndarray and array in NumPy?
  13. Why using an array as an index changes the shape of a multidimensional ndarray?
  14. Python: how to store a numpy multidimensional array in PyTables?
  15. Access n-th dimension in python [duplicate]
  16. How does the axis parameter from NumPy work?
  17. Mapping a NumPy array in place
  18. How to convert a 3d numpy array to 2d
  19. Sorting arrays in NumPy by column
  20. How to access the ith column of a NumPy multidimensional array?
  21. Concatenating two one-dimensional NumPy arrays
  22. Force NumPy ndarray to take ownership of its memory in Cython
  23. How to normalize a NumPy array to within a certain range?
  24. Vectorized NumPy linspace for multiple start and stop values
  25. Compress numpy arrays efficiently
  26. What is the purpose of meshgrid in Python / NumPy?
  27. Index 2D numpy array by a 2D array of indices without loops
  28. Element-wise string concatenation in numpy
  29. Partition array into N chunks with Numpy
  30. Efficient dot products of large memory-mapped arrays

Leave a Comment