Creating a numpy array of 3D coordinates from three 1D arrays

by

To use numpy mesh grid on the above example the following will work:

np.vstack(np.meshgrid(x_p,y_p,z_p)).reshape(3,-1).T

Numpy meshgrid for grids of more then two dimensions require numpy 1.7. To circumvent this and pulling the relevant data from the source code.

def ndmesh(*xi,**kwargs):
    if len(xi) < 2:
        msg = 'meshgrid() takes 2 or more arguments (%d given)' % int(len(xi) > 0)
        raise ValueError(msg)

    args = np.atleast_1d(*xi)
    ndim = len(args)
    copy_ = kwargs.get('copy', True)

    s0 = (1,) * ndim
    output = [x.reshape(s0[:i] + (-1,) + s0[i + 1::]) for i, x in enumerate(args)]

    shape = [x.size for x in output]

    # Return the full N-D matrix (not only the 1-D vector)
    if copy_:
        mult_fact = np.ones(shape, dtype=int)
        return [x * mult_fact for x in output]
    else:
        return np.broadcast_arrays(*output)

Checking the result:

print np.vstack((ndmesh(x_p,y_p,z_p))).reshape(3,-1).T

[[ 1.  2.  8.]
 [ 1.  2.  9.]
 [ 1.  3.  8.]
 ....
 [ 5.  3.  9.]
 [ 5.  4.  8.]
 [ 5.  4.  9.]]

For the above example:

%timeit sol2()
10000 loops, best of 3: 56.1 us per loop

%timeit np.vstack((ndmesh(x_p,y_p,z_p))).reshape(3,-1).T
10000 loops, best of 3: 55.1 us per loop

For when each dimension is 100:

%timeit sol2()
1 loops, best of 3: 655 ms per loop
In [10]:

%timeit points = np.vstack((ndmesh(x_p,y_p,z_p))).reshape(3,-1).T
10 loops, best of 3: 21.8 ms per loop

Depending on what you want to do with the data, you can return a view:

%timeit np.vstack((ndmesh(x_p,y_p,z_p,copy=False))).reshape(3,-1).T
100 loops, best of 3: 8.16 ms per loop

More Related Contents:

  1. Performance of Pandas apply vs np.vectorize to create new column from existing columns
  2. numpy: most efficient frequency counts for unique values in an array
  3. Why is numpy’s einsum faster than numpy’s built in functions?
  4. Numpy sum elements in array based on its value
  5. Efficiently return the index of the first value satisfying condition in array
  6. Frequency counts for unique values in a NumPy array
  7. Most efficient way to forward-fill NaN values in numpy array
  8. Numpy: Fix array with rows of different lengths by filling the empty elements with zeros
  9. Why is a `for` over a Python list faster than over a Numpy array?
  10. Fastest save and load options for a numpy array
  11. Why is Numpy much faster at creating a Zero array compared to replacing the values of an existing array with zeros?
  12. Efficient dot products of large memory-mapped arrays
  13. Store different datatypes in one NumPy array?
  14. Comparing two NumPy arrays for equality, element-wise
  15. Finding indices of matches of one array in another array
  16. How to copy a 2D array into a 3rd dimension, N times?
  17. How do I convert a numpy array to (and display) an image?
  18. How does NumPy’s transpose() method permute the axes of an array?
  19. How do I print the full NumPy array, without truncation?
  20. Why “numpy.any” has no short-circuit mechanism?
  21. Multiplying across in a numpy array
  22. initialize a numpy array
  23. numpy array assignment problem
  24. What does dtype=object mean while creating a numpy array?
  25. Python Numpy TypeError: ufunc ‘isfinite’ not supported for the input types
  26. Selecting close matches from one array based on another reference array
  27. Why does numpy have a corresponding function for many ndarray methods?
  28. Setting the fmt option in numpy.savetxt
  29. Randomly select from numpy array
  30. What is the most efficient way of doing square root of sum of square of two numbers?

Leave a Comment