fancy numpy

1

np.newaxis is None

np.may_share_memory(): check if two arrays share the same memory block

np.nonzero(a): Return the indices of the elements that are non-zero