Array As Input In Python