python处理csv文件绘制曲线_python 读取目录下csv文件并绘制曲线v111

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

"""

Spyder Editor

This temporary script file is located here:

C:\Users\user\.spyder2\.temp.py

"""

"""

Show how to modify the coordinate formatter to report the image "z"

value of the nearest pixel given x and y

V1.1.1 get all filenames in shortout dir choose one to analysis

"""

# coding: utf-8

import time

import string

import os

import math

import pylab

import csv

import numpy as np

from numpy import genfromtxt

import matplotlib

import matplotlib as mpl

from matplotlib.colors import LogNorm

from matplotlib.mlab import bivariate_normal

import matplotlib.pyplot as plt

import matplotlib.cm as cm

import matplotlib.animation as animation

pause = False

linenum=0

fileList=[] #待处理文件路径

for filename in os.listdir(r'D:\shortout'):

pa='D:\shortout\%s'%filename

fileList.append(pa)

#for files in range(0,len(fileList)):

metric = genfromtxt(fileList[0], delimiter=',')

lines=len(metric)

#print len(metric)

#print len(metric[4])

#print metric[4]

rowdatas=metric[:,0]

for index in range(len(metric[4])-1):

a=metric[:,index+1]

rowdatas=np.row_stack((rowdatas,a))

#print len(rowdatas)

#print len(rowdatas[4])

#print rowdatas[4]

#

#plt.figure(figsize=(38,38), dpi=80)

#plt.plot(rowdatas[4] )

#plt.xlabel('time')

#plt.ylabel('value')

#plt.title("USBHID data analysis")

#plt.show()

##如果是参数是list,则默认每次取list中的一个元素,即metric[0],metric[1],...

listdata=rowdatas.tolist()

print listdata[4]

#fig = plt.figure()

#window = fig.add_subplot(111)

#line, = window.plot(listdata[4] )

#plt.ion()

#fig, ax = plt.subplots()

#line, = ax.plot(listdata[4],lw=2)

#ax.grid()

fig = plt.figure()

ax = fig.add_subplot(111)

line, = ax.plot(listdata[4],lw=2 ) # I'm still not clear on this stucture...

ax.grid()

time_template = 'Data ROW = %d'

time_text = ax.text(0.05, 0.9, '', transform=ax.transAxes)

#ax = plt.axes(xlim=(0, 700), ylim=(0, 255))

#line, = ax.plot([], [], lw=2)

def onClick(event):

global pause

pause ^= True

print 'user click the mouse!'

print 'you pressed', event.button, event.xdata, event.ydata

# event.button=1 鼠标左键按下 2 中键按下 3 右键按下

def getData():

global listdata

global linenum

t = 0

while t < len(listdata[4]):

if not pause:

linenum=linenum+1

yield listdata[linenum-1]

# while t < len(listdata[4]):

# t = t + 1

# print t,t

# yield t, t

def update(data):

global linenum

line.set_ydata(data)

time_text.set_text(time_template % (linenum))

return line,

def init():

# ax.set_ylim(0, 1.1)

# ax.set_xlim(0, 10)

# line.set_data(xdata)

plt.xlabel('time')

plt.ylabel('Time')

plt.title('USBHID Data analysis')

return line,

fig.canvas.mpl_connect('button_press_event', onClick)

ani = animation.FuncAnimation(fig, update , getData , blit=False, interval=1*1000,init_func=init,repeat=False)

plt.show()

#my_data = genfromtxt('D:\export.csv', delimiter=',')

#rgbdata=my_data、255

#plt.figure(figsize=(38,38), dpi=80)

#

#for index in range(3):

# row9=rgbdata[:,index]

# print "row %d size is\n"%(index)

# plt.plot(row9 )

# plt.xlabel('time')

# plt.ylabel('value')

# plt.title("USBHID data analysis")

# plt.legend()

## plt.cla()

## plt.clf()

#plt.show()

#plt.figure(1)

#plt.imshow(rgbdata, interpolation='nearest')

#plt.grid(True)

#fig = plt.figure() # 新图 0

#plt.savefig() # 保存

#plt.close('all') # 关闭图 0