Contact
CoCalc Logo Icon
StoreFeaturesDocsShareSupport News Sign UpSign In
| Download

Quick bit of code to plot some data

Views: 56
1
import numpy as np
2
import pandas as pd
3
import matplotlib as mpl
4
#mpl.use('Agg') # Disable X server
5
import matplotlib.pyplot as pyplot
6
import sys
7
from matplotlib import cm
8
9
if len(sys.argv) != 2:
10
print "ERROR!!"
11
print "ERROR!! Command Line Input Error!!"
12
print "ERROR!!"
13
sys.exit()
14
15
csv_file = sys.argv[1]
16
17
csv_check = open(csv_file, 'rb')
18
19
try:
20
a = csv.Sniffer().sniff(csv_check.read(1024))
21
csv_check.seek(0)
22
except csv.Error:
23
24
if ".csv" not in csv_file:
25
print "ERROR!!"
26
print "ERROR!! Improper File Type!!"
27
print "ERROR!!"
28
sys.exit()
29
30
print csv_file
31
32
name = csv_file[csv_file.rfind('/') + 1: -3] + "jpg"
33
print name
34
35
# Read csvfile
36
df = pd.read_csv(csv_file, sep = ' ', names = list('xyz'))
37
38
data = df.pivot('y', 'x', 'z')
39
40
pyplot.xlabel('X')
41
pyplot.ylabel('Y')
42
43
xv = data.columns.values
44
yv = data.index.values
45
46
xt = [xv[0], xv[-1]]
47
yt = [yv[0], yv[-1]]
48
49
pyplot.xticks([0, len(xv) - 1], xt)
50
pyplot.yticks([0, len(yv) - 1], yt)
51
pyplot.imshow(np.log(data.values))
52
pyplot.gca().invert_yaxis()
53
pyplot.savefig(name, dpi=500)
54