EffMap/testMaps.py at master · vicio1975/EffMap · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
# -*- coding: utf-8 -*-
"""
Created on Thu Jul 26 11:53:08 2018

@author: bmusammartanoV
"""

import matplotlib.pyplot as plt
import numpy as np
plt.style.use('seaborn-white')
from scipy import interpolate
from mpl_toolkits.mplot3d import axes3d, Axes3D #<-- Note the capitalization!
from matplotlib.ticker import LinearLocator, FormatStrFormatter
from matplotlib import cm

#import os
#import shutil

#data acquisition
file = "data.csv"
fid = open(file,"r+")
omega = []
Q = []
StatP = []
TotP = []
EffStat = []
EffTot  = []
next(fid) #skip the header file
for line in fid:
    line = line.split(",") #remove the commas
    omega.append(float(line[0]))
    Q.append(float(line[1]))
    StatP.append(float(line[2]))
    TotP.append(float(line[3]))
    EffStat.append(float(line[4]))
    EffTot.append(float(line[5]))
fid.close()

#Plotting the results
omega_ = [2500,2800,3000] #omega values to be considered for the plot and data interpolation
col = ["red","blue","green"]
#Figure for the Efficiency maps
colormaps = ["viridis","plasma","inferno","magma"]
fig1 = plt.figure()
for n,j in enumerate(omega_):
    if j in omega:
        print (j)
        ind = [indx for (indx,x) in enumerate(omega) if x == j  ]
        QQ = Q[ind[0]:ind[-1]+1]
        Tp = TotP[ind[0]:ind[-1]+1]
        plt.plot(QQ,Tp,"-",label=j,color=col[n])
        plt.legend()
        ind = []
        ql = len(QQ)
        QQ = []
        ES = []
    else:
        print("the speed is not on the list!!! moving forward in the list!")

plt.ioff()
ngridx = 300
ngridy = 300
aax = plt.axes()

xp = np.asarray(Q)
yp = np.asarray(TotP)
zp = np.asarray(EffStat)
zp = zp.reshape(ql,len(omega_))

f = interpolate.interp2d(xp,yp,zp, kind='cubic')
x1 = np.linspace(min(Q),max(Q),ngridx)
y1 = np.linspace(1500,max(TotP),ngridy)
z1 = f(x1,y1)
CP = plt.contour(x1, y1, z1,20,cmap=colormaps[0])
plt.clabel(CP, inline=1, fontsize=10, fontcolor="black")
aax.set_xlabel('Q(cms)')
aax.set_ylabel('Total Pressure(Pa)')
plt.show(fig1)

fig2 = plt.figure()
ax = plt.axes(projection='3d')
surf = ax.contour3D(x1, y1, z1 , 200, cmap=colormaps[1])

ax.set_xlabel('Q(cms)')
ax.set_ylabel('Total Pressure(Pa)')
ax.set_zlabel('Eff(-)');
fig2.colorbar(surf, shrink=0.2, aspect=4)

ax.set_zlim(0, 1)
ax.zaxis.set_major_locator(LinearLocator(10))
ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))

plt.show(fig2)

"""
cmaps = [('Perceptually Uniform Sequential', [
            'viridis', 'plasma', 'inferno', 'magma']),
         ('Sequential', [
            'Greys', 'Purples', 'Blues', 'Greens', 'Oranges', 'Reds',
            'YlOrBr', 'YlOrRd', 'OrRd', 'PuRd', 'RdPu', 'BuPu',
            'GnBu', 'PuBu', 'YlGnBu', 'PuBuGn', 'BuGn', 'YlGn']),
         ('Sequential (2)', [
            'binary', 'gist_yarg', 'gist_gray', 'gray', 'bone', 'pink',
            'spring', 'summer', 'autumn', 'winter', 'cool', 'Wistia',
            'hot', 'afmhot', 'gist_heat', 'copper']),
         ('Diverging', [
            'PiYG', 'PRGn', 'BrBG', 'PuOr', 'RdGy', 'RdBu',
            'RdYlBu', 'RdYlGn', 'Spectral', 'coolwarm', 'bwr', 'seismic']),
         ('Qualitative', [
            'Pastel1', 'Pastel2', 'Paired', 'Accent',
            'Dark2', 'Set1', 'Set2', 'Set3',
            'tab10', 'tab20', 'tab20b', 'tab20c']),
         ('Miscellaneous', [
            'flag', 'prism', 'ocean', 'gist_earth', 'terrain', 'gist_stern',
            'gnuplot', 'gnuplot2', 'CMRmap', 'cubehelix', 'brg', 'hsv',
            'gist_rainbow', 'rainbow', 'jet', 'nipy_spectral', 'gist_ncar'])]

"""