[파이썬 연습문제] for문 2차원 리스트 - 2차원리스트 요소의 개수세기

stone은 바둑알이 놓여져 있는 판을 형상화한 것이다.

 

흑돌과 백돌의 개수를 구하기

stone = [[0, 0, 0, 0, 0, 0, 0, 0, 0 ],\
         [0, 1, 0, 1, 2, 1, 2, 1, 0 ],\
         [0, 2, 1, 1, 1, 2, 2, 0, 0 ],\
         [0, 0, 2, 2, 2, 1, 0, 2, 0 ],\
         [0, 0, 0, 0, 0, 1, 0, 2, 1 ],\
         [0, 0, 0, 2, 0, 1, 2, 1, 0 ],\
         [0, 0, 0, 2, 1, 0, 1, 1, 0 ],\
         [0, 0, 0, 1, 1, 0, 0, 0, 0 ],\
         [0, 0, 0, 0, 2, 2, 2, 0, 0 ]]

# 0 = 돌없음
# 1 = 흑돌
# 2 = 백돌

num_black = 0
num_white = 0

for i in range(len(stone)):
    for j in range(len(stone[i])):
        if stone[i][j] == 1:
            num_black += 1
        if stone[i][j] == 2:
            num_white += 1
    
print('흑돌의 개수 : %d' % num_black)
print('백돌의 개수 : %d' % num_white)

흑돌의 개수 : 17
백돌의 개수 : 16

 

 

 

 

출처:  예제 중심 파이썬 입문