AI Assistant를 적용하여 for 루프를 list comprehension으로 변환합니다