| Download
Quick bit of code to plot some data
Project: SCST-CS510-Fall2015
Path: plot_data.py
Views: 56import numpy as np1import pandas as pd2import matplotlib as mpl3#mpl.use('Agg') # Disable X server4import matplotlib.pyplot as pyplot5import sys6from matplotlib import cm78if len(sys.argv) != 2:9print "ERROR!!"10print "ERROR!! Command Line Input Error!!"11print "ERROR!!"12sys.exit()1314csv_file = sys.argv[1]1516csv_check = open(csv_file, 'rb')1718try:19a = csv.Sniffer().sniff(csv_check.read(1024))20csv_check.seek(0)21except csv.Error:2223if ".csv" not in csv_file:24print "ERROR!!"25print "ERROR!! Improper File Type!!"26print "ERROR!!"27sys.exit()2829print csv_file3031name = csv_file[csv_file.rfind('/') + 1: -3] + "jpg"32print name3334# Read csvfile35df = pd.read_csv(csv_file, sep = ' ', names = list('xyz'))3637data = df.pivot('y', 'x', 'z')3839pyplot.xlabel('X')40pyplot.ylabel('Y')4142xv = data.columns.values43yv = data.index.values4445xt = [xv[0], xv[-1]]46yt = [yv[0], yv[-1]]4748pyplot.xticks([0, len(xv) - 1], xt)49pyplot.yticks([0, len(yv) - 1], yt)50pyplot.imshow(np.log(data.values))51pyplot.gca().invert_yaxis()52pyplot.savefig(name, dpi=500)5354