Numpy – create matrix with rows of vector

by

Certainly possible with broadcasting after adding with m zeros along the columns, like so –

np.zeros((m,1),dtype=vector.dtype) + vector

Now, NumPy already has an in-built function np.tile for exactly that same task –

np.tile(vector,(m,1))

Sample run –

In [496]: vector
Out[496]: array([4, 5, 8, 2])

In [497]: m = 5

In [498]: np.zeros((m,1),dtype=vector.dtype) + vector
Out[498]: 
array([[4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2]])

In [499]: np.tile(vector,(m,1))
Out[499]: 
array([[4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2]])

You can also use np.repeat after extending its dimension with np.newaxis/None for the same effect, like so –

In [510]: np.repeat(vector[None],m,axis=0)
Out[510]: 
array([[4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2]])

You can also use integer array indexing to get the replications, like so –

In [525]: vector[None][np.zeros(m,dtype=int)]
Out[525]: 
array([[4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2]])

And finally with np.broadcast_to, you can simply create a 2D view into the input vector and as such this would be virtually free and with no extra memory requirement. So, we would simply do –

In [22]: np.broadcast_to(vector,(m,len(vector)))
Out[22]: 
array([[4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2],
       [4, 5, 8, 2]])

Runtime test –

Here’s a quick runtime test comparing the various approaches –

In [12]: vector = np.random.rand(10000)

In [13]: m = 10000

In [14]: %timeit np.broadcast_to(vector,(m,len(vector)))
100000 loops, best of 3: 3.4 µs per loop # virtually free!

In [15]: %timeit np.zeros((m,1),dtype=vector.dtype) + vector
10 loops, best of 3: 95.1 ms per loop

In [16]: %timeit np.tile(vector,(m,1))
10 loops, best of 3: 89.7 ms per loop

In [17]: %timeit np.repeat(vector[None],m,axis=0)
10 loops, best of 3: 86.2 ms per loop

In [18]: %timeit vector[None][np.zeros(m,dtype=int)]
10 loops, best of 3: 89.8 ms per loop

More Related Contents:

  1. Change sign of elements with an odd sum of indices
  2. How does numpy.newaxis work and when to use it?
  3. How to map numeric data into categories / bins in Pandas dataframe
  4. ImportError: numpy.core.multiarray failed to import
  5. Implement MATLAB’s im2col ‘sliding’ in Python
  6. Installing Numpy on 64bit Windows 7 with Python 2.7.3 [closed]
  7. Convert timedelta64[ns] column to seconds in Python Pandas DataFrame
  8. How do I use np.newaxis?
  9. Numpy: Fix array with rows of different lengths by filling the empty elements with zeros
  10. ImportError: cannot import name NUMPY_MKL
  11. Vectorized NumPy linspace for multiple start and stop values
  12. RuntimeError: module compiled against API version a but this version of numpy is 9
  13. Improve Row Append Performance On Pandas DataFrames
  14. Subtracting numpy arrays of different shape efficiently
  15. NumPy Broadcasting: Calculating sum of squared differences between two arrays
  16. Screen Capture with OpenCV and Python-2.7
  17. Parse a Pandas column to Datetime when importing table from SQL database and filtering rows by date
  18. How to find the pairwise differences between rows of two very large matrices using numpy?
  19. Understanding NumPy’s Convolve
  20. Python pickle protocol choice?
  21. ImportError: DLL load failed when importing Numpy installed in conda virtual environment
  22. Loading text file containing both float and string using numpy.loadtxt
  23. Python Pandas: remove entries based on the number of occurrences
  24. Numpy array broadcasting rules
  25. numpy.sum() giving strange results on large arrays
  26. Elegant way to perform tuple arithmetic
  27. How to implement Garbage Collection in Numpy
  28. Numpy `ValueError: operands could not be broadcast together with shape …`
  29. Python import error: cannot import name ‘six’ from ‘sklearn.externals’
  30. ValueError: x and y must be the same size

Leave a Comment