numpy взять элементы, равные нулю

a[a != 0]
Sleepy Shark