How To Take An Array As Input In Python