Index 2D numpy array by a 2D array of indices without loops

by

When using arrays of indices to index another array, the shape of each index array should match the shape of the output array. You want the column indices to match inds, and you want the row indices to match the row of the output, something like:

array([[0, 0],
       [1, 1],
       [2, 2]])

You can just use a single column of the above, due to broadcasting, so you can use np.arange(3)[:,None] is the vertical arange because None inserts a new axis:

>>> np.arange(3)[:, None]
array([[0],
       [1],
       [2]])

Finally, together:

>>> a[np.arange(3)[:,None], inds]
array([[0, 3],   # a[0,[0,1]]
       [6, 0],   # a[1,[1,2]] 
       [0, 9]])  # a[2,[0,2]]

More Related Contents:

  1. Using numpy to build an array of all combinations of two arrays
  2. How to print the full NumPy array, without truncation?
  3. Convert Python sequence to NumPy array, filling missing values
  4. Numpy array assignment with copy
  5. Working with big data in python and numpy, not enough ram, how to save partial results on disc?
  6. numpy matrix vector multiplication [duplicate]
  7. Repeating each element of a numpy array 5 times
  8. how does multiplication differ for NumPy Matrix vs Array classes?
  9. Find nearest indices for one array against all values in another array – Python / NumPy
  10. Convert a 1D array to a 2D array in numpy
  11. Ambiguity in Pandas Dataframe / Numpy Array “axis” definition
  12. How to split/partition a dataset into training and test datasets for, e.g., cross validation?
  13. How to create a numpy array of arbitrary length strings?
  14. How to extend an array in-place in Numpy?
  15. Numpy modify array in place?
  16. Subsetting a 2D numpy array
  17. How to normalize a 2-dimensional numpy array in python less verbose?
  18. What is the difference between ndarray and array in NumPy?
  19. Removing elements from an array that are in another array
  20. Creating same random number sequence in Python, NumPy and R
  21. Stocking large numbers into numpy array
  22. check for identical rows in different numpy arrays
  23. pd.Timestamp versus np.datetime64: are they interchangeable for selected uses?
  24. Numpy: For every element in one array, find the index in another array
  25. Sort invariant for numpy.argsort with multiple dimensions
  26. Numpy isnan() fails on an array of floats (from pandas dataframe apply)
  27. Concatenate range arrays given start, stop numbers in a vectorized way – NumPy
  28. Numpy – Replace a number with NaN
  29. Python: how to store a numpy multidimensional array in PyTables?
  30. Convert list of lists with different lengths to a numpy array [duplicate]

Leave a Comment