多次元配列の条件を満たす要素の平均を取りたい - numpy

多次元配列の条件を満たす要素の平均を取りたい

同じサイズのbool配列でindexingすると1次元になるので、特定の軸ごとの平均操作などができなくなる

code:py

Acondition.mean(axis=0) # scalar

代わりにmaskとして使って条件を満たさない部分がnp.nanになるようにして、np.nanmean()を使う

code:py

A_masked = np.where(condition, A, np.nan)

A_mean = np.nanmean(A_masked, axis=0) # (*A.shape1:,)