首页 > 代码库 > Python简单做二维统计图

Python简单做二维统计图

先上一张效果图:

技术分享

以上图是一段时间内黄金价格的波动图。

代码如下:

import datetime as DTfrom matplotlib import pyplot as pltfrom matplotlib.dates import date2numdata = []with open("data.txt") as my_file:    for line in my_file:        date, price = line.partition("@")[::2]        data.append((DT.datetime.strptime(date, "%Y-%m-%d %H:%M:%S"), price))d = [date for (date, value) in data[::8]]x = [date2num(date) for (date, value) in data]y = [value for (date, value) in data]fig = plt.figure()graph = fig.add_subplot(111)# Plot the data as a red line with round markers# graph.plot(x, y, ‘r-o‘)graph.plot(x, y)# Set the xtick locations to correspond to just the dates you entered.graph.set_xticks(x)# Set the xtick labels to correspond to just the dates you entered.graph.set_xticklabels(    [date.strftime("%Y-%m-%d %H:%M:%S") for date in d], rotation=30)# plt.grid(True)plt.xticks(x[::8])# print [x[f_value] for f_value in range(0, len(x), 8)]plt.show()

data.txt数据格式如下:

2017-07-29 00:00:02@275672017-07-29 03:00:02@275752017-07-29 06:00:01@275752017-07-29 09:00:01@275752017-07-29 12:00:02@275752017-07-29 15:00:01@275752017-07-29 18:00:01@275752017-07-29 21:00:01@27575

相关知识点介绍:

matplotlib中整个图像是一个Figure对象,在Figure对象中可以包含一个,或者多个Axes对象。每个Axes对象都是一个拥有自己坐标系统的绘图区域,多个Axes对象可以绘成一个比较复杂的图,比如共用x-axis的图。其逻辑关系如下:

技术分享

 

一个具体的图如下:

技术分享

Title为标题。Axis为坐标轴,Label为坐标轴标注。Tick为刻度线,Tick Label为刻度注释,需要注意的是x-axis的ticks(刻度)和x-axis的ticklabels是分开的,ticks就代表x轴的数据,ticklabels表示数据对应的字符串。并不是每个刻度都有字符串对应,ticklabels的密度是可以控制的。往往很密集的刻度会对应合理的字符串便以阅读。

  第一个图的x-axis轴对应的是日期,但是x轴必须有数据,因此matplotlib.dates提供了将日期转化为数据的方法date2num, 这个例子中数据是每3小时有一条,但是显示的时候只到天,具体是如下两行代码:

#每8个取一个日期,其实就是一天d = [date for (date, value) in data[::8]]#每个日期对应一个值,这样才能定位日期的位置,因此值也是每8个取一个plt.xticks(x[::8])

获取x轴和y轴的刻度值

x = [date2num(date) for (date, value) in data]y = [value for (date, value) in data]

创建图像并设置图像位置

fig = plt.figure()graph = fig.add_subplot(111)

111的意思是把figure也就是图像分成1行1列,放在第一个格子,也就是独占整个图像

#把数据画到图上,r是red的意思,线是红色的,o表示对各个值画一个点。# graph.plot(x, y, ‘r-o‘)#默认蓝线不画点graph.plot(x, y)
# Set the xtick labels to correspond to just the dates you entered.#设置x轴label,其实就是上面算好的d日期字符串数组,rotation是label的角度graph.set_xticklabels(    [date.strftime("%Y-%m-%d %H:%M:%S") for date in d], rotation=20)
#图表显示网格plt.grid(True)#设置图标的标题plt.title("Gold price trends")plt.xticks(x[::8])#设置y轴labelplt.ylabel(Gold price/RMB cents)#设置x轴labelplt.xlabel(Date time)#显示图像plt.show()

这么下来一个简单的图表就画好了,很快很实用吧。

参考链接:

https://segmentfault.com/a/1190000006158803

https://matplotlib.org/users/pyplot_tutorial.html

http://bbs.csdn.net/topics/390705164?page=1

Python简单做二维统计图