How To Get An Array As Input In Python From User