时间戳

首页 > 解决方案 > matplotlib 子图 - IndexError：数组索引过多

python - matplotlib 子图 - IndexError：数组索引过多

问题描述

我正在使用subplots函数将 8 列绘制成一个图形。然而，它显示

“IndexError：数组索引过多”

# -*- coding: utf-8 -*-
import pandas as pd
from matplotlib import pyplot as plt
from matplotlib import style

df = pd.read_csv('XXXX', encoding='utf-8')

num = 0

for dim in ['A','B','C','D','E','F','G','H']:
    fig, axes = plt.subplots(nrows=8, ncols=1)
    df[dim].plot(ax=axes[num,0])
    plt.xlabel(dim)
    num += 1

plt.show()

标签： pythonmatplotlibsubplot

解决方案

您的代码有两个问题：

  • 首先，您正在定义subplots()错误的 for 循环内部。您应该只在外面定义一次。
  • 其次，您需要使用axes[num]而不是axes[num, 0]引用特定的子图，因为您只有一列，这就是您获得> IndexError. 如果您有超过 1 列，则 indexing 等将axes[num, 0]起作用。axes[num, 1]

解决方案

# import commands here 

df = pd.read_csv('XXXX', encoding='utf-8')
num = 0

fig, axes = plt.subplots(nrows=8, ncols=1) # <---moved outside for loop

for dim in ['A','B','C','D','E','F','G','H']:
    df[dim].plot(ax=axes[num])
    plt.xlabel(dim)
    num += 1
plt.show()

替代使用enumerate摆脱num变量

fig, axes = plt.subplots(nrows=8, ncols=1)

for i, dim in enumerate(['A','B','C','D','E','F','G','H']):
    df[dim].plot(ax=axes[i])
    plt.xlabel(dim)
plt.show()

推荐阅读