Python绘制CPU曲线图

使用Python绘图工具库，处理和分析系统资源使用情况（本文以CPU使用率为例），且作记录。

1. 工具安装

安装python（此处用的是2.7）及相应的库：matplotlib（windows下需装dateutil、pyparsing、scipy）、numpy；

2. 获取数据

获取进程CPU使用率：

top -d 1 | grep 'test' | grep -v 'grep' >> /tmp/tmp_cpu.log

awk '{print $9}' /tmp/tmp_cpu.log >> /tmp/log/test_cpu.log

3. python绘图

python draw.py test /tmp/log

# -*- coding:utf-8 -*-

import matplotlib.pyplot as pl
import numpy as np
import os
import string
import sys

def set_plot_attr(plt, title, xlabel, ylabel):
plt.title(title)
plt.xlabel(xlabel)
plt.ylabel(ylabel)
plt.grid()
plt.legend()
pl.subplots_adjust(wspace=0.5, hspace=0.5)

def draw_cpu(pro, rootdir):
pro = sys.argv[1]
rootdir = sys.argv[2]
mean_data = []
files = os.listdir(rootdir)
for file in files:
data = []
f = open(rootdir + '/' + file, 'r')
for line in f.readlines():
data.append(string.atof(line.strip()))
f.close()
pl.subplot(211)
pl.plot(data, label=file)
mean_data.append(np.mean(data))

set_plot_attr(pl, pro + ' CPU Usage', 'Time (s)', 'CPU Usage (%)')
pl.subplot(212)
pl.plot(mean_data)
set_plot_attr(pl, pro + ' Mean CPU Usage', 'Time (s)', 'CPU Usage (%)')
pl.show()

def usage():
print 'Usage: draw.py pro rootdir'
print '       pro: progress name'
print '       rootdir: log file directory'

if __name__ == '__main__':
try:
draw_cpu(sys.argv[1], sys.argv[2])
except:
usage()

测试三组数据，结果如下：