Python可视化应用——如何自动控制条形图每根柱子颜色不同

当你使用python脚本绘制条形图和折线图时,有时候并没有给你数据,那么你如何通过代码实现条形图柱子颜色各异呢,以下是我的代码,供大家参考。

import matplotlib

matplotlib.use('Agg')

import numpy as np

import pandas as pd

import matplotlib.pyplot as plt

from matplotlib import mlab

from matplotlib import rcParams

data=[25.9,84.6,21,21.8,36.7,4,5.2,43.7,0.12292,94]#折线图的 Y轴数据

classname=['a','b','c','d','e','f','h','i','j','k']#条形图和折线图的 X轴数据

male=[5,4,7,9,12,3,7,4,10,9]#条形图 Y轴下部分数据

female=[4,6,9,2,7,8,10,8,12,6]#条形图 Y轴上部分数据

#这里数据可以自己设置,也可以在这里加一条def showdata(data,classname,male,female):

#实现数据封装

fig, ax1 = plt.subplots(figsize=(15,10))

plt.xlabel(u'classname')

plt.ylabel(u'male/female')

x=np.arange(len(classname))

y=np.array(list(male))

w=np.array(list(female))

z=np.array(list(data))

xticks1=list(classname)

#tiaoxingtu

colors = []

#定义一个list用来保存colors.py文件内对不同颜色设置的字符

for name, hex in matplotlib.colors.cnames.items():

    colors.append(name)

for i in range(len(classname)):

    rect = plt.bar(x[i],y[i],label='male',align='center',fc = colors[i])

    plt.bar(x[i],w[i],bottom=y[i],label='female',tick_label=y[i],align='center',fc = colors[i+1])

plt.xticks(x,xticks1,size='medium')

plt.legend((rect,),(u"male/female",),loc='upper center',bbox_to_anchor=(0,1))

#这里其实可以用plt.legend(loc='upper center',bbox_to_anchor=(1,1))试一试,就会发现

#有多少根柱子就会显示多少个标签,这样很不方便,所以我就直接显示一个了

for a,b,c in zip(x,y,w):

    plt.text(a, b, '%s' % b, ha='center', va= 'bottom',fontsize=10)

    plt.text(a, c+b, '%s' % c, ha='center', va= 'bottom',fontsize=10)#打印数据

#绘制折线图

ax2 = ax1.twinx()

ax2.plot(x,z,c='y',marker='o',label='data',color='red')

for a,b in zip(x,z):

    plt.text(a, b, '%s' % b, ha='center', label='data',va= 'bottom',fontsize=10)#打印数据

plt.ylabel(u'data')

plt.legend(loc='upper center',bbox_to_anchor=(1,1))

plt.savefig("result.png",format='png')

就这样大功告成啦,一开始,我只是想让每次运行时,不同的柱子颜色不一样就行了,没想到,每次运行这个脚本都会换不同的颜色,下面就是例图啦,每一次运行都会得到不同的图片,嘻嘻,意外之喜。


hhhhh突然觉得很好玩了。

你可能感兴趣的:(Python可视化应用——如何自动控制条形图每根柱子颜色不同)