파이썬(Python) – 리스트의 첫 K개 요소를 N 값으로 교체하는 방법

파이썬에서 리스트의 일부 요소를 특정 값으로 바꿔야 할 때는 복잡한 로직 없이 간단한 반복문만으로 해결할 수 있습니다. 이번 글에서는 리스트의 앞부분 K개 요소를 N 값으로 교체하는 방법을 예제 코드와 함께 살펴보겠습니다.

예제 코드

my_list = [31, 24, 46, 18, 34, 52, 26, 29]
print("리스트 :", my_list)

K = 2
print("K의 값 :", K)

N = 3
print("N의 값 :", N)

for index in range(K):
    my_list[index] = N

print("결과 :", my_list)

실행 결과

리스트 : [31, 24, 46, 18, 34, 52, 26, 29]
K의 값 : 2
N의 값 : 3
결과 : [3, 3, 46, 18, 34, 52, 26, 29]

코드 설명

• 정수 8개로 이루어진 리스트를 정의한 뒤 콘솔에 출력합니다.
• 교체할 요소의 개수를 나타내는 K와 새로 넣을 값을 나타내는 N을 각각 정의하고 출력합니다.
• range(K)를 사용해 인덱스 0부터 K-1까지 순회하면서, 해당 인덱스의 요소에 N 값을 할당합니다.
• 반복이 끝나면 리스트의 첫 K개 요소가 모두 N으로 바뀐 최종 결과가 콘솔에 출력됩니다.

슬라이싱을 활용한 더 간결한 방법

반복문 대신 슬라이싱(slicing)을 사용하면 한 줄로 동일한 작업을 수행할 수 있습니다.

my_list[:K] = [N] * K
print(my_list)  # [3, 3, 46, 18, 34, 52, 26, 29]

[N] * K는 N을 K번 반복한 새 리스트를 만들고, 이를 리스트의 앞 K개 위치에 한 번에 대입합니다. 처리해야 할 요소가 많아질수록 슬라이싱 방식이 더 간결하고 가독성이 좋으므로, 상황에 맞게 두 방식을 선택해 사용하면 됩니다.