matplotlib继承自datetime类图

时间:2012-04-20 01:17:54

标签: python datetime inheritance matplotlib base-class

所以情况是这样的。假设我们有:

class Test(datetime):
    def a(self):
        return '{0:02d}{1:02d}{2:02d}{3:02d}'.format(self.year % 100, self.month, self.day, self.hour)
x = [Test(2010, 05, 02, 06), Test(2010, 05, 02, 12)]
y = [0, 1]

现在尝试使用:

import matplotlib.pyplot as plt
plt.plot(x, y)

给出一个空白图表并出现以下错误:

/usr/local/lib/python2.7/dist-packages/matplotlib/pyplot.py in plot(*args, **kwargs)
   2456         ax.hold(hold)
   2457     try:
-> 2458         ret = ax.plot(*args, **kwargs)
   2459         draw_if_interactive()
   2460     finally:
/usr/local/lib/python2.7/dist-packages/matplotlib/axes.pyc in plot(self, *args, **kwargs)
   3847 
   3848         for line in self._get_lines(*args, **kwargs):
-> 3849             self.add_line(line)
   3850             lines.append(line)
   3851 
/usr/local/lib/python2.7/dist-packages/matplotlib/axes.pyc in add_line(self, line)
   1441             line.set_clip_path(self.patch)
   1442 
-> 1443         self._update_line_limits(line)
   1444         if not line.get_label():
   1445             line.set_label('_line%d'%len(self.lines))
/usr/local/lib/python2.7/dist-packages/matplotlib/axes.pyc in _update_line_limits(self, line)
   1449 
   1450     def _update_line_limits(self, line):
-> 1451         p = line.get_path()
   1452         if p.vertices.size > 0:
   1453             self.dataLim.update_from_path(p, self.ignore_existing_data_limits,
/usr/local/lib/python2.7/dist-packages/matplotlib/lines.pyc in get_path(self)
    642         """
    643         if self._invalidy or self._invalidx:
--> 644             self.recache()
    645         return self._path
    646 
/usr/local/lib/python2.7/dist-packages/matplotlib/lines.pyc in recache(self, always)
    390                 x = ma.asarray(xconv, np.float_)
    391             else:
--> 392                 x = np.asarray(xconv, np.float_)
    393             x = x.ravel()
    394         else:
/usr/local/lib/python2.7/dist-packages/numpy/core/numeric.pyc in asarray(a, dtype, order)
    233 
    234     """
--> 235     return array(a, dtype, copy=False, order=order)
    236 
    237 def asanyarray(a, dtype=None, order=None):
TypeError: float() argument must be a string or a number

你们有谁知道发生了什么事吗?是的,我知道我可以在date2num上使用x,但如果有可能,我想了解错误的原因。

1 个答案:

答案 0 :(得分:2)

Matplotlib使用对象的__class__属性来确定如何处理它。您的Test课程与datetime课程不同:

>>> from datetime import datetime
>>> class Test(datetime):
...   def a(self):
...     return '{0:02d}{1:02d}{2:02d}{3:02d}'.format(self.year % 100, self.month, self.day, self.hour)
...
>>>
>>> t = Test(2010, 05, 02, 06)
>>> d = datetime(2010, 05, 02, 06)
>>>
>>> getattr(t, '__class__')
<class '__main__.Test'>
>>> getattr(d, '__class__')
<type 'datetime.datetime'>

因此,matplotlib无法将您的Test课程视为datetime

这是一个丑陋的黑客,通过注册Test类来解决问题,使其处理方式与datetime相同。有关更多信息,请查看matplotlib.units文档。此代码不保证可以与matplotlib的未来版本一起使用:

>>> import matplotlib.pyplot as plt
>>> import matplotlib.units, matplotlib.dates
>>> import datetime
>>>
>>> class Test(datetime.datetime):
...   def a(self):
...     return '{0:02d}{1:02d}{2:02d}{3:02d}'.format(self.year % 100, self.month, self.day, self.hour)
...
>>> matplotlib.units.registry[Test] = matplotlib.dates.DateConverter()
>>>
>>> x = [Test(2010, 05, 02, 06), Test(2010, 05, 02, 12)]
>>> y = [0, 1]
>>> plt.plot(x, y)
[<matplotlib.lines.Line2D object at 0x02F6E730>]

(我在Windows上使用python 2.7.2和matplotlib 1.1.0测试了这个)