앞의 코드는 다음처럼 수정할 수 있습니다.

def solution(data):
    return [i * i for i in data]

‘주어진 데이터를 변수로 받아 받은 데이터를 → for 문으로 순회하여 리스트 컴프리헨션으로 → 항목마다 데이터를 제곱하고 → 정답을 반환한다’ 형태로 핵심은 그대로 유지하되 중복되는 논리나 단계적인 논리를 하나로 합쳐서 코드의 길이와 불필요한 연산을 줄였습니다. 이렇게 문제를 풀고 난 이후에 자신의 코드를 개선할 점이 있는지 점검하는 게 좋습니다.