How To Get An Array As Input In Python 3