Python에서 2차원 배열 중 특정 열의 값을 모두 바꾸기

Python에서 2차원 배열 중 특정 열의 값을 모두 바꾸기

 

1. Numpy 배열을 사용하는 경우

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
arr[:, 1] = 0

print(arr)

2열의 2,5,8 값을 모두 0으로 변경합니다.

결과

[[1 0 3]
 [4 0 6]
 [7 0 9]]

 

2. 일반 리스트 사용하는 경우

arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

for row in arr:
    row[1] = 0

print(arr)

2열의 2,5,8 값을 모두 0으로 변경합니다.

결과

[[1, 0, 3], [4, 0, 6], [7, 0, 9]]