Python基本图形绘制
对于刚开始接触Python绘图的孩子们,建议利用Python的第三方库matplotlib来学习绘制基本的图形,在此小编整理了绘制折线图、散点图、柱状图三种类型。
小编所用的数据是从网上获取的2017年中国企业500强名单,需要的带走
https://pan.baidu.com/s/1dybjIDyxnsPwwBbO2nr3Pg
密码:vo24
在开始绘制图形之前,给为需要确保所用的工具中装有matplotlib、pandas、pylab三个库,matplotlib自然不必说它的作用了;pandas是用来筛选、处理数据的,可以处理多种格式的数据(.txt,.csv.xls,xlsx)等,可以方便快捷对数据进行清洗;pylab可以帮我我们实现中文显示效果,一般情况下,python不支持中文显示,我们课利用这个库来达到中文显示的效果。
开始绘图
1.折线图
import matplotlib.pyplot as plt
import pandas as pd
from pylab import *
# 折线图
#设置中文显示
mpl.rcParams['font.sans-serif'] = ['SimHei']
#利用pandas读取文件,此路径是当前文件所在的位置,如果文件位置不相同,则修改为文件的绝对路径即可
data = pd.read_csv("500_2.csv",encoding='gb2312')
y1=data['营业收入']
x1=data['企业名称']
#折线图利用.plot()函数绘制
#x1: 表示x轴对应显示的文字
#y1: 表示y轴的高度,即数据值
#label: 表示图注
#linewidth: 表示连线的粗细
#color: 表示颜色
#marker: 表示点的形状
#markerfacecolor: 表示点的内部颜色
#markersize: 表示点的大小
plt.plot(x1,y1,label='Frist line',linewidth=10,color='blue',marker='o',
markerfacecolor='r',markersize=12)
#设置X,Y轴标签
plt.xlabel('Plot Number')
plt.ylabel('Important var')
#设置图标标题
plt.title('Interesting Graph\nCheck it out')
#实现图注,与上方laber配合使用
plt.legend()
#生成的图片保存在本地
plt.savefig('1.png')
#展示
plt.show()
2.散点图
import matplotlib.pyplot as plt
import pandas as pd
from pylab import *
mpl.rcParams['font.sans-serif'] = ['SimHei']
data = pd.read_csv("500_2.csv",encoding='gb2312')
y = data['营业收入']
x = data['企业名称']
#散点图用.scatter()方法,参数和折线图中.plot()相似,不再重复
plt.scatter(x,y,label='frist label',s=20,edgecolors='red')
plt.xlabel('x')
plt.ylabel('y')
plt.title('Interesting Graph\nCheck it out')
plt.legend()
plt.savefig('2.png')
plt.show()
3.柱状图
import matplotlib.pyplot as plt
import pandas as pd
from pylab import *
mpl.rcParams['font.sans-serif'] = ['SimHei']
data = pd.read_csv("500_2.csv",encoding='gb2312')
X = data['企业名称']
Y = data['营业收入']
#此处与以上两个相同
a=plt.bar(X, Y, 0.4, color=("green","red","blue"),)
plt.xlabel("名次")
plt.ylabel("万元")
plt.title("bar")
#此处是利用函数为注定附上相应的值
def autolabel(rects):
for rect in rects:
height = rect.get_height()
plt.text(rect.get_x()+rect.get_width()/2., 1.03*height, '%s' % float(height))
autolabel(a)
plt.savefig('3.png')
plt.show()
到此,python绘图已经结束。以上所述的只是matplotlib绘图功能的冰山一角,matplotlib可以绘制各种图形:热力图,三维图,箱线图,气泡图等等。由于小编水平有限,感兴趣的同学可以查看详细资料,了解更多有关绘图的知识!