Get Array As Input In Python