Cool features of Numpy

a = np.array([6,0,9,0,0,5,0])
def nonzero(a):
    return a!=0
print(a[nonzero(a)])

output:

[6 9 5]