Numpy Array Get row index searching by a row

by

Why not simply do something like this?

>>> a
array([[ 0.,  5.,  2.],
       [ 0.,  0.,  3.],
       [ 0.,  0.,  0.]])
>>> b
array([ 0.,  0.,  3.])

>>> a==b
array([[ True, False, False],
       [ True,  True,  True],
       [ True,  True, False]], dtype=bool)

>>> np.all(a==b,axis=1)
array([False,  True, False], dtype=bool)

>>> np.where(np.all(a==b,axis=1))
(array([1]),)

More Related Contents:

  1. What are the advantages of NumPy over regular Python lists?
  2. Using numpy to build an array of all combinations of two arrays
  3. Find the row indexes of several values in a numpy array
  4. What are the differences between numpy arrays and matrices? Which one should I use?
  5. What is the difference between contiguous and non-contiguous arrays?
  6. Working with big data in python and numpy, not enough ram, how to save partial results on disc?
  7. How to apply a disc shaped mask to a NumPy array?
  8. How to remove specific elements in a numpy array
  9. Count consecutive occurences of values varying in length in a numpy array
  10. Find nearest indices for one array against all values in another array – Python / NumPy
  11. Get the position of the largest value in a multi-dimensional NumPy array
  12. Convert a 1D array to a 2D array in numpy
  13. Using Numpy Vectorize on Functions that Return Vectors
  14. How to extend an array in-place in Numpy?
  15. “In” operator for numpy arrays?
  16. np arrays being immutable – “assignment destination is read-only”
  17. Numpy individual element access slower than for lists
  18. How to turn a video into numpy array?
  19. Can numpy bincount work with 2D arrays?
  20. Sort invariant for numpy.argsort with multiple dimensions
  21. Numpy isnan() fails on an array of floats (from pandas dataframe apply)
  22. Concatenate range arrays given start, stop numbers in a vectorized way – NumPy
  23. Numpy array dimensions
  24. Numpy – Replace a number with NaN
  25. Python: how to store a numpy multidimensional array in PyTables?
  26. Add single element to array in numpy
  27. Numpy shuffle multidimensional array by row only, keep column order unchanged
  28. Randomly select from numpy array
  29. How to “scale” a numpy array?
  30. argsort for a multidimensional ndarray

Leave a Comment