Numpy: Creating a complex array from 2 real ones?

by

This seems to do what you want:

numpy.apply_along_axis(lambda args: [complex(*args)], 3, Data)

Here is another solution:

# The ellipsis is equivalent here to ":,:,:"...
numpy.vectorize(complex)(Data[...,0], Data[...,1])

And yet another simpler solution:

Data[...,0] + 1j * Data[...,1]

PS: If you want to save memory (no intermediate array):

result = 1j*Data[...,1]; result += Data[...,0]

devS’ solution below is also fast.

More Related Contents:

  1. What are the advantages of NumPy over regular Python lists?
  2. ValueError: setting an array element with a sequence
  3. Numpy `logical_or` for more than two arguments
  4. What is the difference between contiguous and non-contiguous arrays?
  5. Quick way to upsample numpy array by nearest neighbor tiling [duplicate]
  6. Count consecutive occurences of values varying in length in a numpy array
  7. Concatenate two NumPy arrays vertically
  8. Getting the indices of several elements in a NumPy array at once
  9. How to do n-D distance and nearest neighbor calculations on numpy arrays
  10. How to conditionally combine two numpy arrays of the same shape
  11. Preserve custom attributes when pickling subclass of numpy array
  12. What does dtype=object mean while creating a numpy array?
  13. Python Numpy TypeError: ufunc ‘isfinite’ not supported for the input types
  14. Why is a `for` over a Python list faster than over a Numpy array?
  15. Selecting close matches from one array based on another reference array
  16. Input and output numpy arrays to h5py
  17. Why does numpy have a corresponding function for many ndarray methods?
  18. Numpy sub-array assignment with advanced, mixed indexing
  19. Efficiently sorting a numpy array in descending order?
  20. Numpy: Divide each row by a vector element
  21. Setting the fmt option in numpy.savetxt
  22. Access n-th dimension in python [duplicate]
  23. numpy division with RuntimeWarning: invalid value encountered in double_scalars
  24. How does the axis parameter from NumPy work?
  25. numpy – evaluate function on a grid of points
  26. numpy arange: how to make “precise” array of floats?
  27. Shift rows of a numpy array independently
  28. Interweaving two NumPy arrays efficiently
  29. Formatting floats in a numpy array [duplicate]
  30. How to convert a 3d numpy array to 2d

Leave a Comment