Getting Array Input In Python From User