如何阻止`colorbar`重塑`networkx`情节? (Python 3)

时间:2016-09-22 18:08:31

标签: python matplotlib colors networkx colorbar

我正在尝试更改colorbar地块上的networkx。当我在那里添加networkx(右)时,条形变得更宽并且还会使我原来的colorbar图(左)消失。

如何让我的colorbar更薄,而不是更改我原来的networkx图表?

下面的代码colorbarhttps://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

1 个答案:

答案 0 :(得分:3)

最简单的方法应该是将颜色条绘制在自己的轴上,并使用[left,bottom,width,height]参数进行游戏。

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