Select One Element in Each Row of a Numpy Array by Column Indices [duplicate]

by

You can choose from given array using numpy.choose which constructs an array from an index array (in your case select_id) and a set of arrays (in your case input_array) to choose from. However you may first need to transpose input_array to match dimensions. The following shows a small example:

In [101]: input_array
Out[101]: 
array([[ 3, 14],
       [12,  5],
       [75, 50]])

In [102]: input_array.shape
Out[102]: (3, 2)

In [103]: select_id
Out[103]: [0, 1, 1]

In [104]: output_array = np.choose(select_id, input_array.T)

In [105]: output_array
Out[105]: array([ 3,  5, 50])

More Related Contents:

  1. Why does Future::select choose the future with a longer sleep period first?
  2. How to check if a value is in the list in selection from pandas data frame?
  3. How to use ng-class in select with ng-options
  4. Select parent element of known element in Selenium
  5. How to select distinct field values using Solr?
  6. How does select work when multiple channels are involved?
  7. MySQL Select all columns from one table and some from another table
  8. Blazor: How to use the onchange event in when using @bind also?
  9. Does a break statement break from a switch/select?
  10. Matplotlib: save plot to numpy array
  11. Find column name in pandas that matches an array
  12. Reshape an array in NumPy
  13. Transform a set of numbers in numpy so that each number gets converted into a number of other numbers which are less than it
  14. NumPy array initialization (fill with identical values)
  15. Slicing of a NumPy 2d array, or how do I extract an mxm submatrix from an nxn array (n>m)?
  16. How can I select from list of values in Oracle
  17. Missing data, insert rows in Pandas and fill with NAN
  18. How do I load a caffe model and convert to a numpy array?
  19. SQL: how to select a single id (“row”) that meets multiple criteria from a single column
  20. How to calculate cumulative normal distribution?
  21. load csv into 2D matrix with numpy for plotting
  22. Create numpy matrix filled with NaNs
  23. How to fix ‘Object arrays cannot be loaded when allow_pickle=False’ for imdb.load_data() function?
  24. Filling in date gaps in MultiIndex Pandas Dataframe
  25. how set numpy floating point accuracy?
  26. How to count RGB or HSV channel combination in an image?
  27. Fitting a Weibull distribution using Scipy
  28. Is there a MATLAB accumarray equivalent in numpy?
  29. Generate random locations within a triangular domain
  30. Why does numpy.histogram (Python) leave off one element as compared to hist in Matlab?

Leave a Comment