How To Get An Array As Input In Python Is Not Working