sum() got an unexpected keyword argument 'where'

Question

I am trying to sum a vector with bool condition with np.sum().

it works on google colab but not on local. Here is my line of code :

np.sum([[0, 1], [np.nan, 5]], where=[False, True], axis=1)

Can anyone help me?

Answer 1

The where=… parameter has been added since numpy-1.17, so you should upgrade the numpy package, for example with:

pip3 install numpy --upgrade