Is there an equivalent to the MATLAB function bsxfun in python?

python c matlab bsxfun
Kevin picture Kevin · Jan 20, 2012 · Viewed 14.6k times · Source

I'm trying to port some of my code from matlab to python, and some of it uses the bsxfun() function for virtual replication followed by multiplication or division (I also use it for logical operations). I'd like to be able to do this without actually replicating the vector (either with a function or with some kind of diagonal matrix) before multiplying or dividing to save on memory and time.

If there's an equivalent of bsxfun in a C library of some kind, that would of course also work.

Answer

capybaralet picture capybaralet · Oct 17, 2014

There isn't really an equivalent of bsxfun, that I'm aware of, although numpy does take care of a lot of broadcasting for you, as others mentioned.

This is commonly touted as an advantage of numpy over matlab, and it is true that a lot of broadcasting is simpler in numpy, but bsxfun is actually more general, because it can take user-defined functions.

Numpy has this: http://docs.scipy.org/doc/numpy/reference/generated/numpy.apply_along_axis.html but only for 1d.

Related questions

Calling C/C++ from Python?

What would be the quickest way to construct a Python binding to a C or C++ library? (I am using Windows if this matters.)

Read More
Convert Python program to C/C++ code?

is it possible to convert a Python program to C/C++? I need to implement a couple of algorithms, and I'm not sure if the performance gap is big enough to justify all the pain I'd go through when doing …

Read More
It is more efficient to use if-return-return or if-else-return?

Suppose I have an if statement with a return. From the efficiency perspective, should I use if(A > B): return A+1 return A-1 or if(A > B): return A+1 else: return A-1 Should I prefer one or another …

Read More