matplotlib + pandas绘图

利用pandas处理日期数据,并根据日期绘制增长率曲线。

处理的json文本内容如下:

# pd.json

[
{
"name": "A", "date": "注册于 2011-6-9"}, {"name": "B", "date": "注册于 2011-6-23"}, {"name": "C", "date": "注册于 2011-7-2"}, {"name": "D", "date": "注册于 2011-10-3"}, {"name": "E", "date": "注册于 2011-8-23"}, {"name": "F", "date": "注册于 2011-9-2"}, {"name": "G", "date": "注册于 2011-10-3"}, {"name": "H", "date": "注册于 2011-11-23"}, {"name": "I", "date": "注册于 2011-12-2"}, {"name": "J", "date": "注册于 2011-12-3"}, {"name": "K", "date": "注册于 2011-12-23"}, {"name": "L", "date": "注册于 2011-10-2"}, {"name": "M", "date": "注册于 2011-8-3"}, {"name": "Y", "date": "注册于 2011-12-23"}, {"name": "U", "date": "注册于 2011-10-9"}, {"name": "W", "date": "注册于 2011-8-6"}, {"name": "K", "date": "注册于 2011-12-21"}, {"name": "L", "date": "注册于 2011-12-5"}, {"name": "M", "date": "注册于 2011-8-3"}, {"name": "Y", "date": "注册于 2011-12-23"}, {"name": "U", "date": "注册于 2011-10-9"}, {"name": "W", "date": "注册于 2011-12-6"} ]

1. 读取json文件的数据(对于自己构建的json文件,里面的键值对字符串必须是双引号

dt = pd.read_json('pd.json', orient='records', encoding='utf-8')

顺便说一下 orient 参数:不一样的文件内容格式,这个参数的值是不一样的,不然会读取失败。

'split' : 索引,列字段,和数据
如: '{"index": [1, 2, 3], "columns": ["a", "b"], "data": [[1, 4], [2, 5], [3, 6]]}'
'records' : 元素为字典的列表
如: '[{"a": 1}, {"b": 2}, {"c": 3}]'
'index' : 索引为键, 值为键值对
如: '{"0": {"a": 3, "b": 5}, "1": {"a": 7, "b": 9}}'
'columns' : 索引为键,值为键值对(注意:输出的数据行列与'index'的不一样
如: '{"a": {"2": 3, "4": 5}, "b": {"6": 7, "8": 9}}',结果与index的例子相同
'values' : 两层的嵌套列表
如: '[["a": 1], ["b": 2]]'

2. 处理日期

# 正则表达式处理
patt = r'(\d{4}-\d{1,2})' # 匹配到月份,也可以匹配到日,再加上 '-\d{1,2}'即可
dt['date'] = dt['date'].str.extract(patt, expand=True)
# 把字符串格式转换为日期格式 dt[
'date'] = pd.to_datetime(dt['date']) # 由于只匹配到月份,to_datetime函数默认日为 01,

此时输出 dt 为:

0  2011-06-01    A
1  2011-06-01    B
2  2011-07-01    C
.....

3. 新增一列 cnt 来统计相同月份的个数,并按日期排序

dl = dt.groupby(['date'], as_index=False)['date'].agg({'cnt': 'count'}).sort_values(by='date')

4. 插入一列 rise 来计算相邻月份总数的差

dl.insert(2, 'rise', (dl['cnt']-dl['cnt'].shift(1))/dl['cnt'].shift(1))

shift()函数对数据进行移动操作,dl['cnt'].shift(1) 表示 dl['cnt'] 列的数据向下移动一个单位

计算差值也可以直接使用 diff() 函数:对数据进行某种移动之后与原数据进行比较得出的差异数据, dl['cnt'].diff()

5. matplotlib绘图

import matplotlib.pyplot as plt

dl.plot(x=dl['date'], y='rise', kind='line', linestyle='-', markerfacecolor='red', marker='o', color='green', figsize=(8, 4), grid=True) plt.show() plt.close()

 

posted @ 2018-12-07 09:25  傲娇的草履虫  阅读(436)  评论(0编辑  收藏  举报