Numpy: Get random set of rows from 2D array

by 
>>> A = np.random.randint(5, size=(10,3))
>>> A
array([[1, 3, 0],
       [3, 2, 0],
       [0, 2, 1],
       [1, 1, 4],
       [3, 2, 2],
       [0, 1, 0],
       [1, 3, 1],
       [0, 4, 1],
       [2, 4, 2],
       [3, 3, 1]])
>>> idx = np.random.randint(10, size=2)
>>> idx
array([7, 6])
>>> A[idx,:]
array([[0, 4, 1],
       [1, 3, 1]])

Putting it together for a general case:

A[np.random.randint(A.shape[0], size=2), :]

For non replacement (numpy 1.7.0+):

A[np.random.choice(A.shape[0], 2, replace=False), :]

I do not believe there is a good way to generate random list without replacement before 1.7. Perhaps you can setup a small definition that ensures the two values are not the same.

More Related Contents:

  1. Binning a column with Python Pandas
  2. Fitting empirical distribution to theoretical ones with Scipy (Python)?
  3. ValueError: setting an array element with a sequence
  4. Numpy `logical_or` for more than two arguments
  5. Best way to join / merge by range in pandas
  6. Concatenate a NumPy array to another NumPy array
  7. ImportError: numpy.core.multiarray failed to import
  8. How do you use the ellipsis slicing syntax in Python?
  9. Fast replacement of values in a numpy array
  10. Concatenate two NumPy arrays vertically
  11. How to make scipy.interpolate give an extrapolated result beyond the input range?
  12. Inverse Distance Weighted (IDW) Interpolation with Python
  13. SQL-like window functions in PANDAS: Row Numbering in Python Pandas Dataframe
  14. numpy array assignment problem
  15. Set values on the diagonal of pandas.DataFrame
  16. What techniques can be used to measure performance of pandas/numpy solutions
  17. numpy: efficiently summing with index arrays
  18. Python Numpy TypeError: ufunc ‘isfinite’ not supported for the input types
  19. RuntimeWarning: numpy.dtype size changed, may indicate binary incompatibility
  20. Find index where elements change value numpy
  21. Why is np.where faster than pd.apply
  22. How do I convert an OpenCV image (BGR and BGRA) to a pygame.Surface object
  23. Why are frequency values rounded in signal using FFT?
  24. 2d convolution using python and numpy
  25. save numpy array in append mode
  26. Converting a 2D numpy array to a structured array
  27. Numpy difference between neighboring elements
  28. new pythonic style for shared axes square subplots in matplotlib?
  29. How to use sklearn fit_transform with pandas and return dataframe instead of numpy array?
  30. numpy array TypeError: only integer scalar arrays can be converted to a scalar index

Leave a Comment