在matplotlib中具有相同数量的xticklabels的六个子图

时间:2013-06-29 10:42:19

标签: python matplotlib axis labels subplot

我真的很难使用matplotlib,尤其是轴设置。我的目标是在一个图中设置6个子图,它们都显示不同的数据集但具有相同数量的ticklabel。

我的源代码的相关部分如下:

graph4.py:

# Import Matolotlib Modules #
import matplotlib as mpl
from matplotlib.figure import Figure
from matplotlib.backends.backend_gtkagg import FigureCanvasGTKAgg as FigureCanvas
from matplotlib import ticker
import matplotlib.pyplot as plt

mpl.rcParams['font.sans-serif']='Arial' #set font to arial 

# Import GTK Modules #

import gtk

#Import System Modules #
import sys

# Import Numpy Modules #
from numpy import genfromtxt
import numpy

# Import Own Modules #
import mysubplot as mysp

class graph4():
    weekdays = ['Montag', 'Dienstag', 'Mittwoch', 'Donnerstag', 'Freitag', 'Samstag']

    def __init__(self, graphview):
        #create new Figure
        self.figure = Figure(figsize=(100,100), dpi=75)

        #create six subplots within self.figure
        self.subplot = []
        for j in range(6):
            self.subplot.append(self.figure.add_subplot(321 + j))


        self.__conf_subplots__() #configure title, xlabel, ylabel and grid of all subplots  


        #to make it look better    
        self.figure.subplots_adjust(left=0.125, bottom=0.1, right=0.9, top=0.96, wspace=0.2, hspace=0.6)    

        #Matplotlib <-> GTK
        self.canvas = FigureCanvas(self.figure) # a gtk.DrawingArea 
        self.canvas.set_flags(gtk.HAS_FOCUS|gtk.CAN_FOCUS)
        self.canvas.grab_focus()
        self.canvas.show()
        graphview.pack_start(self.canvas, True, True)


        #add labels and grid to all subplots  
        def __conf_subplots__(self):
            index = 0
            for i in self.subplot: 
                mysp.conf_subplot(i, 'Zeit', 'Menge', graph4.weekdays[index], True)
                i.plot([], [], 'bo') #empty plot
                index +=1


        def plot(self, filename_list):
            index = 0
            for filename in filename_list:
                data = genfromtxt(filename, delimiter=',') #load data from filename
                if data.size != 0: #only if file isn't empty
                    if index <= len(self.subplot): #plot every file on a different subplot
                        mysp.plot(self.subplot[index],data[0:, 1], data[0:, 0])
                        index +=1


            self.canvas.draw()


            def clear_plot(self):
                #clear axis of all subplots 
                for i in self.subplot:
                    i.cla()

                self.__conf_subplots__() 

mysubplot.py :(辅助模块)

# Import Matplotlib Modules
from matplotlib.axes import Subplot 
import matplotlib.dates as md
import matplotlib.pyplot as plt

# Import Own Modules #
import mytime as myt

# Import Numpy Modules #
import numpy as np

def conf_subplot(subplot, xlabel, ylabel, title, grid):
    if(xlabel != None):
        subplot.set_xlabel(xlabel) 
    if(ylabel != None):
        subplot.set_ylabel(ylabel) 
    if(title != None):
        subplot.set_title(title) 
    subplot.grid(grid)

    #rotate xaxis labels 
    plt.setp(subplot.get_xticklabels(), rotation=30, fontsize=12)

    #display date on xaxis
    subplot.xaxis.set_major_formatter(md.DateFormatter('%H:%M:%S'))      
    subplot.xaxis_date()


def plot(subplot, x, y):
    subplot.plot(x, y, 'bo') 

我认为解释出错的最佳方法是使用屏幕截图。在我开始申请之后,一切看起来都不错:

http://i.imgur.com/SqwaPyG.png

如果我双击左侧的“周”条目,则会调用 graph4.py 中的方法clear_plot()来重置所有子图。然后将文件名列表传递给 graph4.py 中的方法plot()。方法plot()打开每个文件,并在不同的子图上绘制每个数据集。所以在我双击一个条目后,它看起来像:

enter image description here

正如您所看到的,每个子图都有不同数量的xtick标签,这对我来说非常难看。因此,我正在寻找一种解决方案来改善这一点。我的第一种方法是用xaxis.set_ticklabels()手动设置ticklabels,这样每个子图都有相同数量的ticklabel。然而,听起来很奇怪,这只适用于某些数据集,我真的不知道为什么。在一些数据集上,一切正常,在其他数据集上,matplotlib基本上可以做它想要的,并显示我没有指定的xaxis标签。我也尝试了FixedLocator(),但我得到了相同的结果。在某些数据集上它正在工作,而在其他数据集上,matplotlib正在使用不同数量的xtick标签。

我做错了什么?

修改

正如@sgpc建议的那样,我试图使用pyplot。我的源代码现在看起来像这样:

import matplotlib as mpl
import matplotlib.pyplot as plt
from matplotlib.backends.backend_gtkagg import FigureCanvasGTKAgg as FigureCanvas
import matplotlib.dates as md

mpl.rcParams['font.sans-serif']='Arial' #set font to arial 

import gtk
import sys

# Import Numpy Modules #
from numpy import genfromtxt
import numpy

# Import Own Modules #
import mysubplot as mysp

class graph2():
    weekdays = ['Montag', 'Dienstag', 'Mittwoch', 'Donnerstag', 'Freitag', 'Samstag']

    def __init__(self, graphview):
        self.figure, temp = plt.subplots(ncols=2, nrows=3, sharex = True)

        #2d array -> list
        self.axes = [ y for x in temp for y in x]

        #axis: date
        for i in self.axes:
            i.xaxis.set_major_formatter(md.DateFormatter('%H:%M:%S'))
            i.xaxis_date()  

        #make space and rotate xtick labels
        self.figure.autofmt_xdate() 

        #Matplotlib <-> GTK
        self.canvas = FigureCanvas(self.figure) # a gtk.DrawingArea 
        self.canvas.set_flags(gtk.HAS_FOCUS|gtk.CAN_FOCUS)
        self.canvas.grab_focus()
        self.canvas.show()
        graphview.pack_start(self.canvas, True, True)

    def plot(self, filename_list):
        index = 0
        for filename in filename_list:
            data = genfromtxt(filename, delimiter=',') #get dataset
            if data.size != 0: #only if file isn't empty
                if index < len(self.axes): #print each dataset on a different subplot 
                    self.axes[index].plot(data[0:, 1], data[0:, 0], 'bo')
                    index +=1

        self.canvas.draw()

    #not yet implemented
    def clear_plot(self):
        pass

如果我绘制一些数据集,我得到以下输出: http://i.imgur.com/3ngYTNr.png(抱歉,我仍然没有足够的声誉来嵌入图片)

此外,我不确定共享x轴是否是一个非常好的主意,因为有可能x值在每个子图中都不同(例如:在第一个子图中,x值范围从上午8:00 - 上午11:00,在第二个子图中,x值范围从晚上7:00到晚上9:00。

如果我摆脱sharex = True,我会得到以下输出:

http://i.imgur.com/rxHeSyJ.png(抱歉,我仍然没有足够的声誉来嵌入图片)

如您所见,输出现在看起来更好。但是现在,x轴上的标签不会更新。我认为那是因为最后的情节是空的。

我的下一次尝试是为每个子图使用一个轴。因此,我做了这个改变:

for i in self.axes:
    plt.setp(i.get_xticklabels(), visible=True, rotation = 30) #<-- I added this line...
    i.xaxis.set_major_formatter(md.DateFormatter('%H:%M:%S'))
    i.xaxis_date() 

#self.figure.autofmt_xdate() #<--changed this line
self.figure.subplots_adjust(left=0.125, bottom=0.1, right=0.9, top=0.96, wspace=0.2, hspace=0.6) #<-- and added this line

现在我得到以下输出:

i.imgur.com/TmA1goE.png(抱歉,我仍然没有足够的声誉嵌入图片)

因此,通过此次尝试,我基本上遇到了与Figure()add_subplot()相同的问题。

我真的不知道,还有什么我可以尝试让它发挥作用......

1 个答案:

答案 0 :(得分:0)

我强烈建议您将pyplot.subplots()sharex=True

一起使用
fig, axes = subplots(ncols=2, nrows=3, sharex= True)

然后使用以下方法访问每个轴:

ax = axes[i,j]

你可以策划:

ax.plot(...)

要控制您可以使用的每个AxesSubplot的刻度数:

ax.locator_params(axis='x', nbins=6)

OBS:axis可以是'x''y''both'