Python Pattern Match Array Function