1

pcolor を使用した matplotlib のヒートマップpcolorの例を使用してヒートマップを作成しようとしていますか? しかし、私は困っています。

最初に、受け入れられた anwser の例が機能せず、次のエラー メッセージが表示されます。

Traceback (most recent call last):
  File "/home/knorrs/temp/HeatMapTest.py", line 22, in <module>
    heatmap = ax.pcolor(nba_sort, cmap=plt.cm.Blues, alpha=0.8)
  File "/home/martin/pybin/lib/python2.7/site-packages/matplotlib/axes.py", line 7309, in pcolor
    X, Y, C = self._pcolorargs('pcolor', *args)
  File "/home/martin/pybin/lib/python2.7/site-packages/matplotlib/axes.py", line 7132, in _pcolorargs
    numRows, numCols = C.shape
AttributeError: 'NoneType' object has no attribute 'shape'

「私の」方法でプロットを作成すると、1 つの例外を除いて良い結果が得られます。y 軸のラベル/目盛りはしばらくすると停止します。y 軸のラベルの数は、x 軸のラベルの数と同じでなければならないようです。

これは、ヒートマップの作成を担当するコードの一部です。data_test2 次元の numpy 配列 ( data_test.shapeyields ) であることに注意してください(20, 13)。

import matplotlib.pyplot as pl
import numpy as np

def plot_heatmap(data):
    x_min = ((data.shape[1]-1)/2)*-1    #if the shape is for example 13 (has to be odd) we set the x_min to -6
    x_max = (data.shape[1]-1)/2         #and the x_max to +6
    x_labels = range(x_min,x_max+1,1)   #this way we create the x_labels going from -6 over 0 to +6

    fig = pl.figure(figsize=(24,18))
    ax = fig.add_subplot(1,1,1)
    plot = ax.pcolor(data, cmap=pl.cm.Blues, edgecolors='k')

    # put the major ticks at the middle of each cell
    ax.set_xticks(np.arange(data.shape[0])+0.5, minor=False)
    ax.set_yticks(np.arange(data.shape[1])+0.5, minor=False)
    ax.set_ybound(lower = 0, upper = data.shape[0])
    ax.set_xbound(lower = 0, upper = data.shape[1])

    # want a more natural, table-like display
    ax.invert_yaxis()
    #ax.xaxis.tick_top()

    ax.set_xticklabels(x_labels, minor=False)
    ax.set_yticklabels(AA, minor=False)
    fig.colorbar(plot)

    pl.show()

AA = ['G', 'A', 'V', 'S', 'T', 'C', 'M', 'L', 'I', 'K', 'R', 'E', 'D', 'Q', 'N', 'F', 'Y', 'W', 'P', 'H']

data_test = np.random.rand(20,13)
plot_heatmap(data_test)

結果のプロット

この動作の理由と変更方法を教えていただければ幸いです。

4

1 に答える 1

3

この動作は、コードの次の行が原因です。

ax.set_yticks(np.arange(data.shape[1])+0.5, minor=False)

間違った次元のデータで yticks を定義しています!

次のように修正する場合:

ax.set_xticks(np.arange(data.shape[1])+0.5, minor=False)
ax.set_yticks(np.arange(data.shape[0])+0.5, minor=False)

それが動作します。

于 2013-08-27T11:13:50.703 に答える