Coolog 🖥️

/

[Python] 2차원 리스트 생성, 입력, 찾기


💫 2차원 리스트란?

matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]
  • 리스트 안에 또 다른 리스트가 들어있는 구조
  • 보통 행(row)과 열(column) 형태의 데이터를 표현할 때 사용

💫 2차원 리스트 생성하기

# 직접 초기화
matrix = [
    [0, 0, 0],
    [0, 0, 0],
    [0, 0, 0]
]

# List Comprehension 활용
rows, cols = 3, 4
matrix = [[0 for _ in range(cols)] for _ in range(rows)]
print(matrix)  # [[0,0,0,0],[0,0,0,0],[0,0,0,0]]

💫 값 입력하기

# 특정 위치 값 변경
matrix[1][2] = 99
print(matrix)  # 두 번째 행, 세 번째 열이 99로 바뀜

# 반복문으로 채우기
rows, cols = 3, 3
matrix = [[0 for _ in range(cols)] for _ in range(rows)]

num = 1
for i in range(rows):
    for j in range(cols):
        matrix[i][j] = num
        num += 1

print(matrix)  # [[1,2,3],[4,5,6],[7,8,9]]

💫 값 찾기

# 반복문으로 찾기
match_list = []
target = 1

for i in range(len(matrix)):
    for j in range(len(matrix[i])):
        if matrix[i][j] == target:
            match_list.append([i,j])

# List Comprehension으로 위치 찾기
match_list = [(i, j) for i in range(len(matrix)) for j in range(len(matrix[i])) if matrix[i][j] == target]
print(match_list)  # [(1, 1)]

답글 남기기

이메일 주소는 공개되지 않습니다. 필수 필드는 *로 표시됩니다

최근 게시물