Numpy select rows based on condition

by

Use a boolean mask:

mask = (z[:, 0] == 6)
z[mask, :]

This is much more efficient than np.where because you can use the boolean mask directly, without having the overhead of converting it to an array of indices first.

One liner:

z[z[:, 0] == 6, :]

More Related Contents:

  1. What are the differences between numpy arrays and matrices? Which one should I use?
  2. numpy matrix vector multiplication [duplicate]
  3. Numpy matrix to array
  4. how does multiplication differ for NumPy Matrix vs Array classes?
  5. Convert a 1D array to a 2D array in numpy
  6. how to perform max/mean pooling on a 2d array using numpy
  7. Why does numpy.linalg.solve() offer more precise matrix inversions than numpy.linalg.inv()?
  8. NumPy array/matrix of mixed types
  9. Convert row vector to column vector in NumPy
  10. Is there a NumPy function to return the first index of something in an array?
  11. Understanding NumPy’s einsum
  12. Intuition and idea behind reshaping 4D array to 2D array in NumPy
  13. Computing the correlation coefficient between two multi-dimensional arrays
  14. Removing nan values from an array
  15. Why is numpy’s einsum faster than numpy’s built in functions?
  16. How to return 0 with divide by zero
  17. Reshape an array in NumPy
  18. Most efficient way to forward-fill NaN values in numpy array
  19. Why does numpy.zeros takes up little space
  20. Efficiently Calculating a Euclidean Distance Matrix Using Numpy
  21. How to split a numpy array in fixed size chunks with and without overlap?
  22. Parameters to numpy’s fromfunction
  23. List to array conversion to use ravel() function
  24. Efficient distance calculation between N points and a reference in numpy/scipy
  25. Fastest save and load options for a numpy array
  26. Numpy find indices of groups with same value
  27. numpy with python: convert 3d array to 2d
  28. How is numpy’s fancy indexing implemented?
  29. How can I slice each element of a numpy array of strings?
  30. Efficient dot products of large memory-mapped arrays

Leave a Comment