Using Numpy Vectorize on Functions that Return Vectors

by

np.vectorize is just a convenience function. It doesn’t actually make code run any faster. If it isn’t convenient to use np.vectorize, simply write your own function that works as you wish.

The purpose of np.vectorize is to transform functions which are not numpy-aware (e.g. take floats as input and return floats as output) into functions that can operate on (and return) numpy arrays.

Your function f is already numpy-aware — it uses a numpy array in its definition and returns a numpy array. So np.vectorize is not a good fit for your use case.

The solution therefore is just to roll your own function f that works the way you desire.

More Related Contents:

  1. Subtracting numpy arrays of different shape efficiently
  2. Concatenate range arrays given start, stop numbers in a vectorized way – NumPy
  3. How to print the full NumPy array, without truncation?
  4. Convert Python sequence to NumPy array, filling missing values
  5. numpy matrix vector multiplication [duplicate]
  6. Repeating each element of a numpy array 5 times
  7. how does multiplication differ for NumPy Matrix vs Array classes?
  8. Bin elements per row – Vectorized 2D Bincount for NumPy
  9. Vectorized searchsorted numpy
  10. How to split/partition a dataset into training and test datasets for, e.g., cross validation?
  11. How to create a numpy array of arbitrary length strings?
  12. How to extend an array in-place in Numpy?
  13. Vectorized look-up of values in Pandas dataframe
  14. “In” operator for numpy arrays?
  15. Efficient evaluation of a function at every cell of a NumPy array
  16. np arrays being immutable – “assignment destination is read-only”
  17. vectorize conditional assignment in pandas dataframe
  18. Numpy individual element access slower than for lists
  19. How to turn a video into numpy array?
  20. Can numpy bincount work with 2D arrays?
  21. Sort invariant for numpy.argsort with multiple dimensions
  22. Numpy isnan() fails on an array of floats (from pandas dataframe apply)
  23. Efficiently replace elements in array based on dictionary – NumPy / Python
  24. Numpy – Replace a number with NaN
  25. Python: how to store a numpy multidimensional array in PyTables?
  26. Numpy shuffle multidimensional array by row only, keep column order unchanged
  27. Randomly select from numpy array
  28. How to “scale” a numpy array?
  29. Numpy Broadcast to perform euclidean distance vectorized
  30. Numpy `ValueError: operands could not be broadcast together with shape …`

Leave a Comment