Is there a multi-dimensional version of arange/linspace in numpy?

by

You can use np.mgrid for this, it’s often more convenient than np.meshgrid because it creates the arrays in one step:

import numpy as np
X,Y = np.mgrid[-5:5.1:0.5, -5:5.1:0.5]

For linspace-like functionality, replace the step (i.e. 0.5) with a complex number whose magnitude specifies the number of points you want in the series. Using this syntax, the same arrays as above are specified as:

X, Y = np.mgrid[-5:5:21j, -5:5:21j]

You can then create your pairs as:

xy = np.vstack((X.flatten(), Y.flatten())).T

As @ali_m suggested, this can all be done in one line:

xy = np.mgrid[-5:5.1:0.5, -5:5.1:0.5].reshape(2,-1).T

Best of luck!

More Related Contents:

  1. Cartesian product of x and y array points into single array of 2D points
  2. Using numpy to build an array of all combinations of two arrays
  3. Fast punctuation removal with pandas
  4. Efficiently Using Multiple Numpy Slices for Random Image Cropping
  5. How to one-hot-encode from a pandas column containing a list?
  6. Computing the correlation coefficient between two multi-dimensional arrays
  7. Why is numpy’s einsum faster than numpy’s built in functions?
  8. Using a pre-trained word embedding (word2vec or Glove) in TensorFlow
  9. Numpy index slice without losing dimension information
  10. numpy: formal definition of “array_like” objects?
  11. Select multiple ranges of columns in Pandas DataFrame
  12. How to convert list of numpy arrays into single numpy array?
  13. how to perform max/mean pooling on a 2d array using numpy
  14. How to find the index of a value in 2d array in Python?
  15. Filtering (reducing) a NumPy Array
  16. How can I upgrade NumPy?
  17. How to replicate array to specific length array
  18. Zero pad numpy array
  19. How to install matplotlib with Python3.2
  20. Plotting implicit equations in 3d
  21. Cannot get minor grid lines to appear in matplotlib figure
  22. Weird behaviour initializing a numpy array of string data
  23. Extract csv file specific columns to list in Python
  24. How to “scale” a numpy array?
  25. Resample time series in pandas to a weekly interval
  26. How to convert an image from np.uint16 to np.uint8?
  27. Python import error: cannot import name ‘six’ from ‘sklearn.externals’
  28. Passing C++ vector to Numpy through Cython without copying and taking care of memory management automatically
  29. Numpy Broadcast to perform euclidean distance vectorized
  30. ValueError: x and y must be the same size

Leave a Comment