Take An Array Input In Python