np.where with broadcasting

[rjenc29]

Small extension to np.where to support use cases like:

a = np.linspace(-2, 5, 20).reshape(4, 5)
cond = a > 0
mask = np.where(cond, 1, 0)

Initial commit for CI.

[codecov-io]

Codecov Report

❗ No coverage uploaded for pull request base (master@7d8fca1). Click here to learn what that means.
The diff coverage is 82%.

@@           Coverage Diff            @@
##             master   #3630   +/-   ##
========================================
  Coverage          ?   80.5%           
========================================
  Files             ?     395           
  Lines             ?   81117           
  Branches          ?    9234           
========================================
  Hits              ?   65300           
  Misses            ?   14389           
  Partials          ?    1428

[rjenc29]

Hi, let me know what you think of this relatively simple extension. Could arguably be taken further either in this PR or subsequently; might have incremental utility on its own.

[stuartarchibald]

Thanks for the patches. Am I right in thinking that this is with view of closing #3339 ?

[rjenc29]

I hadn't spotted that one to be honest, but it should fit the bill. It was more of a thought experiment which turned into a PR :)

[stuartarchibald]

This is pending update RE #3650

[rjenc29]

The last commit should extend support to cover some basic use cases.

Let me know what you think, ta!

[stuartarchibald]

Thanks for the patch, adding support for these additional cases is really useful. Looks good to merge.