File qt_plot.py changed (mode: 100644) (index bc2cd5f..a9da129) |
... |
... |
class SimpleSimplexEstimationGraph(pg.PlotWidget): |
936 |
936 |
|
|
937 |
937 |
|
|
938 |
938 |
def export_to_excel(self): |
def export_to_excel(self): |
939 |
|
pass |
|
|
939 |
|
#č já bych nechtěl, aby mně export najednou spadl |
|
940 |
|
#č z jakéhokoliv důvodu |
|
941 |
|
try: |
|
942 |
|
proposal_filename = self.dice_box.guessbox.filename |
|
943 |
|
if proposal_filename: |
|
944 |
|
proposal_filename += '.xlsx' |
|
945 |
|
else: |
|
946 |
|
proposal_filename = self.dice_box.gm_signature + '.xlsx' |
|
947 |
|
filename, *__ = pg.FileDialog.getSaveFileName(self, 'Export to Excel',\ |
|
948 |
|
proposal_filename, initialFilter='*.xlsx') |
|
949 |
|
self.df.to_excel(filename) |
|
950 |
|
except BaseException as e: |
|
951 |
|
print(self.__class__.__name__ + ":", repr(e)) |
|
952 |
|
|
|
953 |
|
|
940 |
954 |
|
|
941 |
955 |
|
|
942 |
956 |
|
|
|
... |
... |
class SimpleSimplexEstimationGraph(pg.PlotWidget): |
1063 |
1077 |
|
|
1064 |
1078 |
|
|
1065 |
1079 |
# it can be effectively done with pandas |
# it can be effectively done with pandas |
1066 |
|
df = pd.DataFrame(tuple(tri_estimation.values())) |
|
|
1080 |
|
self.df = df = pd.DataFrame(tuple(tri_estimation.values())) |
1067 |
1081 |
# -1 = 'out', 0=success, 1=failure, 2=mix |
# -1 = 'out', 0=success, 1=failure, 2=mix |
1068 |
1082 |
df.rename(columns={-1:'out', 0:'success', 1:'failure', 2:'mix'}, inplace=True) |
df.rename(columns={-1:'out', 0:'success', 1:'failure', 2:'mix'}, inplace=True) |
1069 |
1083 |
df.insert(loc=0, column='x', value=tuple(tri_estimation.keys()), allow_duplicates=False) |
df.insert(loc=0, column='x', value=tuple(tri_estimation.keys()), allow_duplicates=False) |
|
... |
... |
class TriEstimationGraph(SimpleSimplexEstimationGraph): |
1104 |
1118 |
data = self.dice_box.guessbox.estimations[self.tri_estimation_name] |
data = self.dice_box.guessbox.estimations[self.tri_estimation_name] |
1105 |
1119 |
x = data[0] |
x = data[0] |
1106 |
1120 |
# it can be effectively done with pandas |
# it can be effectively done with pandas |
1107 |
|
df = pd.DataFrame(data[1]) |
|
|
1121 |
|
self.df = df = pd.DataFrame(data[1]) |
1108 |
1122 |
# -1 = 'out', 0=success, 1=failure, 2=mix |
# -1 = 'out', 0=success, 1=failure, 2=mix |
1109 |
1123 |
df.rename(columns={-1:'out', 0:'success', 1:'failure', 2:'mix'}, inplace=True) |
df.rename(columns={-1:'out', 0:'success', 1:'failure', 2:'mix'}, inplace=True) |
1110 |
1124 |
|
|
|
... |
... |
class TriEstimationGraph(SimpleSimplexEstimationGraph): |
1112 |
1126 |
|
|
1113 |
1127 |
self.pen_f.setData(x, self.zerosafe(df.failure)) |
self.pen_f.setData(x, self.zerosafe(df.failure)) |
1114 |
1128 |
self.pen_outmix.setData(x, self.zerosafe(df.failure+df.mix)) |
self.pen_outmix.setData(x, self.zerosafe(df.failure+df.mix)) |
1115 |
|
self.pen_success.setData(x, self.zerosafe(df.failure+df.mix+df.out)) # kontrolu, zda je to 1, nechame uživateli |
|
|
1129 |
|
self.pen_success.setData(x, self.zerosafe(df.failure+df.mix+df.out)) |
1116 |
1130 |
|
|
1117 |
1131 |
if 'vertex_estimation' in self.dice_box.guessbox.estimations: |
if 'vertex_estimation' in self.dice_box.guessbox.estimations: |
1118 |
1132 |
data = self.dice_box.guessbox.estimations['vertex_estimation'] |
data = self.dice_box.guessbox.estimations['vertex_estimation'] |