A Function With An Input Array In Python