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

소개

matplotlib에서 작업하다 보면 종종 'TypeError: 'Axes' object is not subscriptable'라는 오류에 부딪힐 수 있습니다. 이는 아마도 Axes 객체를 리스트처럼 인덱스 접근을 시도했기 때문일 것입니다. 이번 블로그 글에서는 이 오류가 발생하는 원인과 해결 방법을 자세히 살펴보겠습니다.

에러 발생 예시 코드

먼저, 'TypeError: 'Axes' object is not subscriptable' 에러를 유발할 수 있는 간단한 예시 코드를 확인해봅시다.

import matplotlib.pyplot as plt

# 새로운 플롯 생성
fig, ax = plt.subplots()

# Axes 객체에 인덱스 접근 시도
first_plot = ax[0].plot([1, 2, 3], [1, 4, 9])
plt.show()

에러 해결 방법

1. Axes 객체의 직접적인 접근 방식을 수정

matplotlib의 Axes 객체는 리스트가 아니기 때문에 인덱스 접근을 사용할 수 없습니다. 대신, Axes 객체에 plotting 메서드를 직접 호출해야 합니다.

import matplotlib.pyplot as plt

# 새로운 플롯 생성
fig, ax = plt.subplots()

# Axes 객체에 접근하여 플롯 그리기
ax.plot([1, 2, 3], [1, 4, 9])
plt.show()

2. 여러 개의 서브플롯 사용 시 인덱스 문제 해결

서브플롯을 생성할 경우, Axes 객체를 2차원 배열처럼 접근해야 할 때가 있습니다. 이를 위해 다음과 같은 형식의 코드를 사용할 수 있습니다.

import matplotlib.pyplot as plt

# 2x2 서브플롯 생성
fig, axs = plt.subplots(2, 2)

# 각 서브플롯에 접근
axs[0, 0].plot([1, 2, 3], [1, 4, 9])
axs[0, 1].plot([1, 2, 3], [9, 4, 1])
plt.show()

마무리

이번 블로그 글에서는 matplotlib에서 발생할 수 있는 'TypeError: 'Axes' object is not subscriptable' 에러에 대한 해결 방법을 살펴보았습니다. Axes 객체에 대한 인덱스 접근을 피하고, 적절한 메서드를 사용함으로써 이 오류를 극복할 수 있습니다. matplotlib을 사용할 때는 항상 문서를 참조하며 작업을 진행하는 것이 좋은 습관입니다!