matplotlib TypeError: 'AxesSubplot' object is not subscriptable 오류 해결하기

소개

Python의 matplotlib 라이브러리를 사용하다 보면 가끔 'TypeError: 'AxesSubplot' object is not subscriptable' 에러에 부딪히기도 합니다. 이 오류는 주로 AxesSubplot 객체에서 인덱스를 사용하려 할 때 발생합니다. 오늘은 이 오류의 원인과 간단한 해결 방법을 알아보겠습니다.

에러 발생 예시 코드

우선, 이 에러가 발생할 수 있는 간단한 예시 코드를 살펴보겠습니다.

import matplotlib.pyplot as plt

# 그림과 축 생성
fig, ax = plt.subplots()

# 그래프 생성
ax.plot([1, 2, 3], [1, 4, 9])

# 'AxesSubplot' 객체를 잘못 인덱스 접근
title = ax[0].set_title('Sample Title')
plt.show()

에러 해결 방법

1. 축 객체에 대한 인덱스 접근을 하지 않기

AxesSubplot 객체는 단일 객체이기 때문에 인덱스를 사용하여 접근할 수 없습니다. 축의 속성에 직접 접근하여 조작해야 합니다. 예를 들어, 다음과 같이 수정할 수 있습니다.

import matplotlib.pyplot as plt

# 그림과 축 생성
fig, ax = plt.subplots()

# 그래프 생성
ax.plot([1, 2, 3], [1, 4, 9])

# Title 설정
ax.set_title('Sample Title')
plt.show()

2. 여러 서브플롯을 사용할 때 인덱스 활용

서브플롯을 생성하여 다수의 AxesSubplot 객체를 사용할 경우, 인덱스를 사용해야 하나, 이 경우 axes를 사용하여 리스트 형태로 액세스할 필요가 있습니다.

import matplotlib.pyplot as plt

# 여러 개의 서브플롯 생성
fig, axes = plt.subplots(2, 2)

# 각 서브플롯에 데이터 플로팅
axes[0, 0].plot([1, 2, 3], [1, 4, 9])
axes[0, 0].set_title('First Plot')

axes[0, 1].plot([1, 2, 3], [2, 3, 5])
axes[0, 1].set_title('Second Plot')

plt.tight_layout()
plt.show()

마무리

이 블로그 글에서는 matplotlib을 사용할 때 발생할 수 있는 'TypeError: 'AxesSubplot' object is not subscriptable' 에러에 대한 원인과 해결 방법을 살펴보았습니다. 인덱스 접근을 피하고, AxesSubplot 객체의 속성들을 직접 활용함으로써 문제를 해결할 수 있습니다. 앞으로 matplotlib로 멋진 데이터 비주얼을 만드는 여정이 되길 바랍니다!