Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Python에서 리스트 요소를 K번 반복해 확장하는 방법

개요

이 글에서는 Python에서 리스트의 각 요소를 K번 반복하여 새로운 리스트로 확장하는 방법을 알아봅니다. 두 가지 서로 다른 방법으로 이 문제를 해결할 수 있습니다.

방법 1: 반복 연산자(*) 사용하기

다음 단계에 따라 문제를 해결할 수 있습니다.

  • 리스트, K, 그리고 결과를 담을 빈 리스트를 초기화합니다.
  • 리스트를 순회하면서 반복 연산자(*)를 사용해 현재 요소를 K번 추가합니다.
  • 결과를 출력합니다.

예제 코드

# 리스트 초기화
numbers = [1, 2, 3]
K = 5

# 빈 리스트 생성
result = []

# 리스트 확장하기
for i in numbers:
    result += [i] * K

# 결과 출력
print(result)

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

출력 결과

[1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3]

방법 2: 리스트 컴프리헨션으로 내부 루프 활용하기

다음 단계에 따라 문제를 해결할 수 있습니다.

  • 리스트와 K를 초기화합니다.
  • 리스트 컴프리헨션 안에서 내부 루프를 사용해 현재 요소를 K번 반복합니다.
  • 결과를 출력합니다.

예제 코드

# 리스트 초기화
numbers = [1, 2, 3]
K = 5

# 리스트 확장하기
result = [i for i in numbers for j in range(K)]

# 결과 출력
print(result)

위 코드를 실행하면 첫 번째 방법과 동일한 결과가 출력됩니다.

출력 결과

[1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3]

마무리

지금까지 Python에서 리스트의 각 요소를 K번 반복해 확장하는 두 가지 방법을 살펴보았습니다. 반복 연산자를 사용하는 방식은 직관적이고 이해하기 쉬운 반면, 리스트 컴프리헨션을 활용하는 방식은 한 줄로 간결하게 작성할 수 있다는 장점이 있습니다. 코드의 가독성과 상황에 맞게 적절한 방법을 선택해 사용하시기 바랍니다. 궁금한 점이 있다면 댓글로 남겨주세요.