Python Array Apply Function