Open nj-vs-vh opened 2 hours ago
Also, it might be interesting that on a large dataset the manual vectorized operation (normalize weights by their row-wise sum, multiply data by them and sum row-wise) is much faster compared with ak.mean(data, weight, axis=1)
. For my dataset the latter is ~10 sec and the former is <0.1 sec.
Version of Awkward Array
2.6.9
Description and code to reproduce
I am doing a row-wise weighted mean in an awkward array and getting wrong results.
Here's an MRE with outputs in the comments: