Python Apply Function To Array