Repeat each values of an array different times

by

That’s exactly what numpy.repeat does:

>>> a = np.array([0.1, 0.2, 0.3, 0.4, 0.5, 0.6])
>>> s = np.array([3, 3, 9, 3, 6, 3])
>>> np.repeat(a, s)
array([ 0.1,  0.1,  0.1,  0.2,  0.2,  0.2,  0.3,  0.3,  0.3,  0.3,  0.3,
        0.3,  0.3,  0.3,  0.3,  0.4,  0.4,  0.4,  0.5,  0.5,  0.5,  0.5,
        0.5,  0.5,  0.6,  0.6,  0.6])

In pure Python you can do something like:

>>> from itertools import repeat, chain, imap
>>> list(chain.from_iterable(imap(repeat, a, s)))
[0.1, 0.1, 0.1, 0.2, 0.2, 0.2, 0.3, 0.3, 0.3, 0.3, 0.3, 0.3, 0.3, 0.3, 0.3, 0.4, 0.4, 0.4, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.6, 0.6, 0.6]

But of course it is going to be way slower than its NumPy equivalent:

>>> s = [3, 3, 9, 3, 6, 3]*1000
>>> a = [0.1, 0.2, 0.3, 0.4, 0.5, 0.6]*1000
>>> %timeit list(chain.from_iterable(imap(repeat, a, s)))
1000 loops, best of 3: 1.21 ms per loop
>>> %timeit np.repeat(a_a, s_a) #a_a and s_a are NumPy arrays of same size as a and b
10000 loops, best of 3: 202 µs per loop

More Related Contents:

  1. Sorting arrays in NumPy by column
  2. Performance of Pandas apply vs np.vectorize to create new column from existing columns
  3. How to access the ith column of a NumPy multidimensional array?
  4. Python/NumPy first occurrence of subarray
  5. Concatenating two one-dimensional NumPy arrays
  6. Numpy matrix to array
  7. Replacing Numpy elements if condition is met
  8. How to make a multidimension numpy array with a varying row size?
  9. Frequency counts for unique values in a NumPy array
  10. How can I add new dimensions to a Numpy array?
  11. Force NumPy ndarray to take ownership of its memory in Cython
  12. How to normalize a NumPy array to within a certain range?
  13. subsampling every nth entry in a numpy array
  14. How do I create an empty array and then append to it in NumPy?
  15. Immutable numpy array?
  16. Check if two 3D numpy arrays contain overlapping 2D arrays
  17. Numpy: Creating a complex array from 2 real ones?
  18. Binary random array with a specific proportion of ones?
  19. Add multiple values to one numpy array index
  20. Index 2D numpy array by a 2D array of indices without loops
  21. Flattening a list of NumPy arrays?
  22. Element-wise string concatenation in numpy
  23. Fastest pairwise distance metric in python
  24. Partition array into N chunks with Numpy
  25. Numpy Vector (N,1) dimension -> (N,) dimension conversion
  26. numpy with python: convert 3d array to 2d
  27. How is numpy’s fancy indexing implemented?
  28. Convert row vector to column vector in NumPy
  29. Passing numpy arrays in Cython to a C function that requires dynamically allocated arrays
  30. Convert list of lists with different lengths to a numpy array [duplicate]

Leave a Comment