How To Get An Array As Input In Python