2016-09-22 7 views
0

私のnetworkxプロットのcolorbarを変更しようとしています。私がcolorbarをそこに(右)追加すると、バーが余分に広がって、私の元のnetworkxプロット(左)をsmooshes。`colorbar`が` networkx`プロットを再形成するのを止めるには? (Python 3)

colorbarをシンナーにして、元のnetworkxグラフを変更しないにはどうすればよいですか?

私のコードはcolorbarで、https://stackoverflow.com/users/5285918/laneryですが、大きなネットワークで使用しました。

# Set up Graph 
DF_adj = pd.DataFrame(np.array(
    [[1, 0, 1, 1], 
    [0, 1, 1, 0], 
    [1, 1, 1, 1], 
    [1, 0, 1, 1] ]), columns=['sepal length (cm)', 'sepal width (cm)', 'petal length (cm)', 'petal width (cm)'], index=['sepal length (cm)', 'sepal width (cm)', 'petal length (cm)', 'petal width (cm)']) 

G = nx.Graph(DF_adj.as_matrix()) 
G = nx.relabel_nodes(G, dict(zip(range(4), ['sepal length (cm)', 'sepal width (cm)', 'petal length (cm)', 'petal width (cm)']))) 

# Color mapping 
color_palette = sns.cubehelix_palette(3) 
cmap = {k:color_palette[v-1] for k,v in zip(['sepal length (cm)', 'sepal width (cm)', 'petal length (cm)', 'petal width (cm)'],[2, 1, 3, 2])} 

# Draw 
fig, ax = plt.subplots() 
nx.draw(G, node_color=[cmap[node] for node in G.nodes()], with_labels=True, ax=ax) 

sm = plt.cm.ScalarMappable(cmap=ListedColormap(color_palette), 
          norm=plt.Normalize(vmin=0, vmax=3)) 
sm._A = [] 
fig.colorbar(sm, ticks=range(1,4)) 

enter image description here

答えて

3

最も簡単な方法は、自身の軸の上にカラーバーをプロットし、[左、下、幅、高さ]パラメータをいじっしなければなりません。

cbaxes = fig.add_axes([0.9, 0.1, 0.015, 0.8]) 
sm = plt.cm.ScalarMappable(cmap=ListedColormap(color_palette), 
          norm=plt.Normalize(vmin=0, vmax=3)) 
sm._A = [] 
plt.colorbar(sm, cax=cbaxes, ticks=range(4)) 

出典:positioning the colorbar

関連する問題