Inputting An Array In Python