I 'think' I understand what you mean but looks like more changes than I would feel comfortable doing at the moment given my lack of in-depth understanding of Numpy  (I wrote code about what I knew for sure while perhaps missing other scenarios, but changing the sense would imply I know every time it 'might') .