[Python] 프로그래머스 배열의 원소만큼 추가하기

문제 설명

아무 원소도 들어있지 않은 빈 배열 X가 있습니다. 양의 정수 배열 arr가 매개변수로 주어질 때, arr의 앞에서부터 차례대로 원소를 보면서 원소가 a라면 X의 맨 뒤에 a를 a번 추가하는 일을 반복한 뒤의 배열 X를 return 하는 solution 함수를 작성해 주세요.

 

https://school.programmers.co.kr/learn/courses/30/lessons/181861

프로그래머스

제한 사항

1 ≤ arr의 길이 ≤ 100
1 ≤ arr의 원소 ≤ 100

예제

arr	result
[5, 1, 4]	[5, 5, 5, 5, 5, 1, 4, 4, 4, 4]
[6, 6]	[6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6]
[1]	[1]

문제 풀이

def solution(arr):
    answer = []
    
    for ele in arr:
        for _ in range(ele):
            answer.append(ele)
            
    return answer

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.