using an numpy array as indices of the 2nd dim of another array? [duplicate]

by

You can use NumPy's purely integer array indexing –

A[np.arange(A.shape[0]),B.ravel()]

Sample run –

In [57]: A
Out[57]: 
array([[0, 1],
       [2, 3],
       [4, 5]])

In [58]: B
Out[58]: 
array([[1],
       [0],
       [1]])

In [59]: A[np.arange(A.shape[0]),B.ravel()]
Out[59]: array([1, 2, 5])

Please note that if B is a 1D array or a list of such column indices, you could simply skip the flattening operation with .ravel().

Sample run –

In [186]: A
Out[186]: 
array([[0, 1],
       [2, 3],
       [4, 5]])

In [187]: B
Out[187]: [1, 0, 1]

In [188]: A[np.arange(A.shape[0]),B]
Out[188]: array([1, 2, 5])

More Related Contents:

  1. How to extract an arbitrary line of values from a numpy array?
  2. Get the position of the largest value in a multi-dimensional NumPy array
  3. Why does indexing numpy arrays with brackets and commas differ in behavior?
  4. Increment Numpy array with repeated indices
  5. Generalise slicing operation in a NumPy array
  6. Find index where elements change value numpy
  7. Numpy: For every element in one array, find the index in another array
  8. Index all *except* one item in python
  9. Numpy sub-array assignment with advanced, mixed indexing
  10. How is numpy’s fancy indexing implemented?
  11. Access n-th dimension in python [duplicate]
  12. numpy array TypeError: only integer scalar arrays can be converted to a scalar index
  13. Cython: (Why / When) Is it preferable to use Py_ssize_t for indexing?
  14. Get the indices of N highest values in an ndarray
  15. Select elements of numpy array via boolean mask array
  16. Numpy where function multiple conditions
  17. How to implement the Softmax function in Python
  18. What is the inverse function of zip in python? [duplicate]
  19. In-place type conversion of a NumPy array
  20. Multiprocessing.Pool makes Numpy matrix multiplication slower
  21. Extracting specific columns in numpy array
  22. How to get element-wise matrix multiplication (Hadamard product) in numpy?
  23. How to get the indices list of all NaN value in numpy array?
  24. Python’s and Numpy’s nan and set
  25. Weird behaviour initializing a numpy array of string data
  26. How to “scale” a numpy array?
  27. Python import error: cannot import name ‘six’ from ‘sklearn.externals’
  28. Passing C++ vector to Numpy through Cython without copying and taking care of memory management automatically
  29. Numpy Broadcast to perform euclidean distance vectorized
  30. ValueError: x and y must be the same size

Leave a Comment