paulin/EasyGA
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

General graph features

Browse Source
This commit is contained in:
SimpleArt
2020-11-12 12:11:04 -05:00
parent 555c187e39
commit 357df1609f
1 changed files with 9 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/database/matplotlib_graph.py
View File

@ -4,6 +4,11 @@ import matplotlib.pyplot as plt
class Matplotlib_Graph: class Matplotlib_Graph:
"""Prebuilt graphing functions to make visual represention of fitness data.""" """Prebuilt graphing functions to make visual represention of fitness data."""
type_of_plot_dict = {
'line' : plt.plot,
'scatter' : plt.scatter,
'bar' : plt.bar
}
def __init__(self, database): def __init__(self, database):
self.database = database self.database = database
@ -119,15 +124,8 @@ class Matplotlib_Graph:
@type_of_plot.setter @type_of_plot.setter
def type_of_plot(self, value_input): def type_of_plot(self, _type_of_plot):
if _type_of_plot in self.type_of_plot_dict.keys():
# Defults type of ploting functions self._type_of_plot = self.type_of_plot_dict[_type_of_plot]
if(value_input == "line"):
self._type_of_plot = plt.plot
elif(value_input == "scatter"):
self._type_of_plot = plt.scatter
elif(value_input == "bar"):
self._type_of_plot = plt.bar
else: else:
# If its none of the defaults then use what the user provided. self._type_of_plot = _type_of_plot
self._type_of_plot = value_input