NumPy Broadcasting: Calculating sum of squared differences between two arrays

by

You can use np.einsum after calculating the differences in a broadcasted way, like so –

ab = a[:,None,:] - b
out = np.einsum('ijk,ijk->ij',ab,ab)

Or use scipy's cdist with its optional metric argument set as 'sqeuclidean' to give us the squared euclidean distances as needed for our problem, like so –

from scipy.spatial.distance import cdist
out = cdist(a,b,'sqeuclidean')

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. Numpy – create matrix with rows of vector
  4. How do I use np.newaxis?
  5. Vectorized NumPy linspace for multiple start and stop values
  6. Subtracting numpy arrays of different shape efficiently
  7. How to find the pairwise differences between rows of two very large matrices using numpy?
  8. Numpy array broadcasting rules
  9. Numpy `ValueError: operands could not be broadcast together with shape …`
  10. Is there a NumPy function to return the first index of something in an array?
  11. Understanding NumPy’s einsum
  12. Saving a Numpy array as an image
  13. How to one-hot-encode from a pandas column containing a list?
  14. Removing nan values from an array
  15. Numpy index slice without losing dimension information
  16. Numpy: Get random set of rows from 2D array
  17. Select multiple ranges of columns in Pandas DataFrame
  18. How can I release memory after creating matplotlib figures
  19. Sliding window of M-by-N shape numpy.ndarray
  20. Improve Row Append Performance On Pandas DataFrames
  21. Precision lost while using read_csv in pandas
  22. Improving FFT performance in Python
  23. Parameters to numpy’s fromfunction
  24. Creating a Confidence Ellipses in a sccatterplot using matplotlib
  25. How to generate audio from a numpy array?
  26. Python Pandas – Missing required dependencies [‘numpy’] 1
  27. Distance between point and a line (from two points)
  28. ImportError: No module named numpy on spark workers
  29. Automatic White Balancing with Grayworld assumption
  30. C++ conversion from NumPy array to Mat (OpenCV)

Leave a Comment