How To Take An Array Input In Python