Matplotlib hexbin添加颜色条 - 为假人

时间:2016-11-27 20:00:44

标签: python matplotlib colorbar

无法弄清楚如何简单地将颜色条添加到hexbin matplotlib图表的矩阵中。这是一个7乘7的图表矩阵,这是第一行代码的一部分:

#importing nec. libs
import pandas as pd
from pylab import rcParams
from matplotlib import pyplot as PLT
%matplotlib inline

#reading data
dataset_cat_coded_path = 'C:/Users/IBM_ADMIN/Documents/Sberbank/dataset_cat_coded.csv'
dataset_cat_coded = pd.read_csv(dataset_cat_coded_path, encoding = 'cp1251')
test_frame = dataset_cat_coded[['opf','priority', 'sales_method_id', 'is_block_vks','sector_id', 'segment_id', 'org_type_id']]



#building figure
rcParams['figure.figsize']=30,30 #fig. size

#fig. initiation
fig = PLT.figure()

#setting parameters defining min and max luminance
vmin = 0
vmax=test_frame.shape[0]

#building a row of plots
ax1 = fig.add_subplot(7,7,2)
ax1.hexbin(test_frame.opf, test_frame.priority,C=None,gridsize=12,bins=None,mincnt=1, vmin = vmin, vmax=vmax)
ax1.set_title('priority')
ax1.set_ylabel('OPF')

ax2 = fig.add_subplot(7,7,3)
ax2.hexbin(test_frame.opf, test_frame.sales_method_id,C=None,gridsize=12,bins=None,mincnt=1, vmin = vmin, vmax=vmax)
ax2.set_title('sales_method_id')

ax3 = fig.add_subplot(7,7,4)
ax3.hexbin(test_frame.opf, test_frame.is_block_vks,C=None,gridsize=12,bins=None,mincnt=1, vmin = vmin, vmax=vmax)
ax3.set_title('is_block_vks')

ax4 = fig.add_subplot(7,7,5)
ax4.hexbin(test_frame.opf, test_frame.sector_id,C=None,gridsize=12,bins=None,mincnt=1, vmin = vmin, vmax=vmax)
ax4.set_title('sector_id')

ax5 = fig.add_subplot(7,7,6)
ax5.hexbin(test_frame.opf, test_frame.segment_id,C=None,gridsize=12,bins=None,mincnt=1, vmin = vmin, vmax=vmax)
ax5.set_title('segment_id')

ax6 = fig.add_subplot(7,7,7)
ax6.hexbin(test_frame.opf, test_frame.org_type_id,C=None,gridsize=12,bins=None,mincnt=1, vmin = vmin, vmax=vmax)
ax6.set_title('org_type_id')

#colorbar attempt
cbar_ax = fig.add_axes([0.85, 0.15, 0.05, 0.7])
fig.colorbar(ax6, cax=cbar_ax, location='bottom') here

错误文字:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-17-e885eb3cd39a> in <module>()
     36 
     37 cbar_ax = fig.add_axes([0.85, 0.15, 0.05, 0.7])
---> 38 fig.colorbar(ax6, cax=cbar_ax, location='bottom')

C:\Program Files\Anaconda3\lib\site-packages\matplotlib\figure.py in colorbar(self, mappable, cax, ax, use_gridspec, **kw)
   1591                 cax, kw = cbar.make_axes(ax, **kw)
   1592         cax.hold(True)
-> 1593         cb = cbar.colorbar_factory(cax, mappable, **kw)
   1594 
   1595         self.sca(current_ax)

C:\Program Files\Anaconda3\lib\site-packages\matplotlib\colorbar.py in colorbar_factory(cax, mappable, **kwargs)
   1328         cb = ColorbarPatch(cax, mappable, **kwargs)
   1329     else:
-> 1330         cb = Colorbar(cax, mappable, **kwargs)
   1331 
   1332     cid = mappable.callbacksSM.connect('changed', cb.on_mappable_changed)

C:\Program Files\Anaconda3\lib\site-packages\matplotlib\colorbar.py in __init__(self, ax, mappable, **kw)
    878         # Ensure the given mappable's norm has appropriate vmin and vmax set
    879         # even if mappable.draw has not yet been called.
--> 880         mappable.autoscale_None()
    881 
    882         self.mappable = mappable

AttributeError: 'AxesSubplot' object has no attribute 'autoscale_None'

结果情节: enter image description here

1 个答案:

答案 0 :(得分:0)

正如我在评论中指出的那样,this question显示了如何添加颜色条。

在您的代码中

fig.colorbar(ax6, cax=cbar_ax, location='bottom')

您尝试将颜色条设置为图形,而不指定它应该与哪个对象相关联。因此,您会收到一个错误,它基本上告诉您没有为mappable提供任何对象。

您需要做的是将您的hexbin图像提供给颜色条,以便它知道实际绘制哪些颜色。

import numpy as np
import matplotlib.pyplot as plt

n = 1000
x = np.random.standard_normal(n)
y = 2.0 + 3.0 * x + 4.0 * np.random.standard_normal(n)

fig, axes = plt.subplots(nrows=2, ncols=2)
for ax in axes.flat:
    im = ax.hexbin(x,y)

fig.subplots_adjust(right=0.8)
cbar_ax = fig.add_axes([0.85, 0.15, 0.05, 0.7])
fig.colorbar(im, cax=cbar_ax)

plt.show()