How To Get Array As Input In Python