생태계 문제 풀기

import numpy as np
import copy

matrix = np.array(
    [
        [0, 1, 1, 1, 1, 0, 0, 1, 0, 0],
        [0, 0, 1, 0, 1, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 1, 0, 1, 0],
        [0, 0, 0, 0, 0, 1, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 1, 1, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
    ]
)

matrix_copy = copy.deepcopy(matrix)

# 매트릭스를 돌아다닌다
# 값이 1이라면 -> 행 인덱스 가져옴 -> 같은 열에 있는 값 추가

for i in range(len(matrix)):
    for j in range(len(matrix[i])):
        if matrix_copy[i][j] == 1:
            matrix[i] += matrix[j]

print("직접 간접적 피식자의 합 matrix *********")

for row in matrix:
    for count in row:
        print(count, end=" ")
    print()

print(
    f"""
생물별 직접 간접적 피식자의 합 *********

독수리: {sum(matrix[0])}
뱀: {sum(matrix[1])}
쥐: {sum(matrix[2])}
휘파람새: {sum(matrix[3])}
개구리: {sum(matrix[4])}
나방: {sum(matrix[5])}
메뚜기: {sum(matrix[6])}
다람쥐: {sum(matrix[7])}
과실: {sum(matrix[8])}
나무: {sum(matrix[9])}"""
)