Why not just ``return a[a[2] > 0]``?