-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathmyPlot.py
139 lines (122 loc) · 4.74 KB
/
myPlot.py
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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
'''
This library is used to plot the fuzzification results and the correlation between attributes after the fuzzification process.
'''
import matplotlib.pyplot as plt
import seaborn as sns
path_1 = ''
path_2 = ''
def garis(x, y):
'''
This function is to plot a line as a marker on
X coordinates from Y=0 to Y=y.
Parameter:
x = coordinate X
y = maximum line height desired
Outputs:
Plot the dotted line as the X-coordinate marker.
'''
plt.plot([x, x], [0, y], 'k--')
def garis_atas(x, y):
'''
This function is to plot a line as a marker at coordinate X
from Y=y to Y=1.
Parameter:
x = coordinate X
y = minimum desired line height
Outputs:
Plot the dotted line as the X-coordinate marker.
'''
plt.plot([x, x], [y, 1], 'k--')
def mf (x, y, kategori, t_tengah, judul, kode, xlbl, titik,
y_garis_bawah=[1], y_garis_atas=[0.56], y_text=[0.5],
pisah_garis=[False], rot=[0], rot_xtick=0,
simpan=True, path=path_1, lang='id'):
'''
Fungsi ini untuk plot himpunan fuzzy dengan fungsi keanggotaan dan
range nilai yang diberikan.
Parameter:
x: range nilai pada sb. X
y: list fungsi keanggotaan
kategori: kategori untuk setiap fungsi keanggotaan
t_tengah: koordinat X untuk setiap kategori
judul: judul plot dan sekaligus sebagai tambahan pada nama file jika disimpan
kode: tambahan kode untuk nama file jika ingin menyimpan hasil plot
xlbl: keterangan sumbu X
titik: kumpulan titik sebagai untuk membantu memperjelas fungsi keanggotaan
y_garis_bawah: untuk garis penanda titik, sebagai parameter Y untuk
fungsi `garis` (default)=[1])
y_garis_atas: untuk garis penanda titik, sebagai parameter Y untuk
fungsi `garis_atas` (default=[0.56])
y_text: koordinat Y pada setiap kategori dari fungsi keanggotaan yang diberikan
(default=[0.5])
pisah_garis: parameter boolean untuk memisahkan garis vertikal penanda koordinat X
pada setiap titik (default=[False])
rot: untuk merotasi kategori (default=[0])
rot_xtick: untuk merotasi titik pada sb. X (default=0)
simpan: untuk menyimpan langsung hasil plot (default=False)
path: directory untuk menyimpan hasil plot
Output:
Line plot himpunan fuzzy.
'''
plt.close()
sns.set_style('white')
warna = ['r', 'g', 'b','deeppink', 'k', 'm', 'c', 'darkorange', 'gray', 'chocolate']*10
plt.figure(figsize=(10,6))
if y_garis_bawah == [1]:
y_garis_bawah=[1]*(len(titik)-2)
if y_garis_atas == [0.56]:
y_garis_atas=[0.56]*(len(titik)-2)
if pisah_garis == [False]:
pisah_garis = [False]*(len(titik)-2)
if y_text == [0.5]:
y_text = [0.5]*(len(kategori))
if rot == [0]:
rot = [0]*(len(kategori))
for i in range(len(kategori)):
plt.plot(x, y[i], warna[i], linewidth=2)
plt.text(t_tengah[i], y_text[i], kategori[i], fontsize=18.5, rotation=rot[i]) #-len(kategori[i])/20
plt.title(judul, fontsize=18)
plt.ylim(0,1.05)
plt.xlim(x[0],x[-1])
for u in range(1, len(titik)-1):
if pisah_garis[u-1] is False:
garis(titik[u], y_garis_bawah[u-1])
else:
garis_atas(titik[u], y_garis_atas[u-1])
garis(titik[u], y_garis_bawah[u-1])
plt.tick_params(which='major', axis='both', labelsize=14)
plt.tick_params(axis='x', rotation=rot_xtick) #baru
plt.xticks(titik)
if lang=='id':
plt.ylabel('Derajat Keanggotaan', fontsize=16)
else:
plt.ylabel('Degree of Membership', fontsize=16)
plt.xlabel(xlbl, fontsize=16)
plt.savefig(path+judul+kode+'.jpg', dpi=300)
# plt.savefig(path_2+judul+kode+'.jpg', dpi=300)
def korelasi(judul, datanya, warna, kode='', simpan=True, path=path_1):
'''
Fungsi ini untuk membuat plot heatmap korelasi dari data yang diberikan.
Parameter:
judul: judul untuk heatmap dan juga sebagai kode untuk nama file jika ingin disimpan langsung
datanya: sumber data yang ingin dibuat heatmapnya
warna: palette warna yang diinginkan untuk heatmap
kode: kode sebagai tambahan nama file jika ingin disimpan langsung
simpan: untuk menyimpan file (Default=False)
path: path/directory untuk menyimpan hasil heatmpat (Default='')
Output:
Plot heatmap dari data yang diberikan.
'''
plt.close()
plt.rcParams['figure.figsize'] = (16,10)
g = sns.heatmap(datanya, annot=True, linewidths=.5, cmap=warna, center=0, vmin=-1, vmax=1, annot_kws={"size": 16})
g.set_xticklabels(g.get_xmajorticklabels(), fontsize = 16, rotation=45, ha='right')
g.set_yticklabels(g.get_ymajorticklabels(), fontsize = 16, rotation=45)
cbar = g.collections[0].colorbar
cbar.ax.tick_params(labelsize=18)
plt.title(judul, fontsize = 30)
plt.tight_layout()
plt.savefig(path_1 +judul +kode +'.jpg', dpi=300)
# plt.savefig(path_2 +'\\' +judul +'.png', dpi=300)
plt.show()
plt.close()