Vectorized way of calculating row-wise dot product two matrices with Scipy

by

Straightforward way to do that is:

import numpy as np
a=np.array([[1,2,3],[3,4,5]])
b=np.array([[1,2,3],[1,2,3]])
np.sum(a*b, axis=1)

which avoids the python loop and is faster in cases like:

def npsumdot(x, y):
    return np.sum(x*y, axis=1)

def loopdot(x, y):
    result = np.empty((x.shape[0]))
    for i in range(x.shape[0]):
        result[i] = np.dot(x[i], y[i])
    return result

timeit npsumdot(np.random.rand(500000,50),np.random.rand(500000,50))
# 1 loops, best of 3: 861 ms per loop
timeit loopdot(np.random.rand(500000,50),np.random.rand(500000,50))
# 1 loops, best of 3: 1.58 s per loop

More Related Contents:

  1. numpy elementwise outer product
  2. Python (NumPy, SciPy), finding the null space of a matrix
  3. Show confidence limits and prediction limits in scatter plot
  4. What does “an intermediate result is being cached” mean?
  5. Efficiently Creating A Pandas DataFrame From A Numpy 3d array
  6. Fitting to Poisson histogram
  7. Converting image from RGB to HSV color space
  8. Taking subarrays from numpy array with given stride/stepsize
  9. Sorting arrays in NumPy by column
  10. Moving average or running mean
  11. Save / load scipy sparse csr_matrix in portable data format
  12. Find matching rows in 2 dimensional numpy array
  13. Resampling a numpy array representing an image
  14. Vectorized NumPy linspace for multiple start and stop values
  15. python recursive vectorization with timeseries
  16. Get coordinates of local maxima in 2D array above certain value
  17. Python baseline correction library
  18. How to get faster code than numpy.dot for matrix multiplication?
  19. How to get element-wise matrix multiplication (Hadamard product) in numpy?
  20. NumPy version of “Exponential weighted moving average”, equivalent to pandas.ewm().mean()
  21. How to share numpy random state of a parent process with child processes?
  22. Generating Discrete random variables with specified weights using SciPy or NumPy
  23. Fastest pairwise distance metric in python
  24. How to get the indices list of all NaN value in numpy array?
  25. setting spacing between grouped bar plots in matplotlib
  26. Compute a confidence interval from sample data assuming unknown distribution
  27. Is the order of a Python dictionary guaranteed over iterations?
  28. Is there a numpy/scipy dot product, calculating only the diagonal entries of the result?
  29. sampling random floats on a range in numpy
  30. Fast interpolation of grid data

Leave a Comment