이 코드 역시 변수를 사용하지 않고도 작성할 수 있습니다. 변수를 사용하지 않으면 다소 복잡하지만, 놀랍게도 코드가 세 줄에서 한 줄로 줄어듭니다.
s_mon = [0, 0, 0]
for row in data:
if row[-1] == '전표매입':
s_mon[int(row[0].split('-')[1]) - 10] += int(row[-3])
print(s_mon)
앞의 코드에 이어 과정 3의 그래프를 그려 봅시다.
import matplotlib.pyplot as plt
s_mon = [0, 0, 0]
for row in data:
if row[-1] == '전표매입':
mon, payment = int(row[0].split('-')[1]), int(row[-3])
idx = mon - 10
s_mon[idx] += payment
plt.rc('font', family='Malgun Gothic')
plt.title('10~12월 지출현황')
plt.bar(['10월', '11월', '12월'], s_mon, color='royalblue')
plt.show()