janishutz/BiogasControllerApp
1
0
Fork 0
mirror of https://github.com/janishutz/BiogasControllerApp.git synced 2025-11-25 05:44:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

fit.py: Update to improve syntax

Browse Source
This commit is contained in:
Janis Hutz
2025-05-26 17:16:31 +02:00
parent 096003ffa9
commit f560c24574
1 changed files with 30 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
plot_generator/fit.py
View File

@@ -1,36 +1,29 @@
import csv
import numpy as np
import matplotlib.pyplot as plt
go = 0
n = int(input("Aktuelle Sondennummer: "))
n = int(input("Sensor number to be printed: "))
try:
imp = open("Sonden2021.csv", "r")
go = 1
except FileNotFoundError:
print("Failed to open file (non-existent or corrupted?)")
go = 0
file = ""
if go == 1:
reader = csv.reader(imp, delimiter=',')
rohdaten = list(reader)
rohdaten.sort(key=lambda imp: float(imp[2]))
lenght = len(rohdaten)
def generate_plot():
reader = csv.reader(file, delimiter=',')
data = list(reader)
data.sort(key=lambda imp: float(imp[2]))
lenght = len(data)
x = []
y = []
for i in range(lenght):
extract = rohdaten.pop(0)
sondennummer = int(extract.pop(0))
if sondennummer == n:
for _ in range(lenght):
extract = data.pop(0)
sensor = int(extract.pop(0))
if sensor == n:
ye = extract.pop(0)
xe = extract.pop(0)
y.append(float(ye))
x.append(float(xe))
fit = np.polyfit(x, y, 2)
print(fit)
formula = f"F(U) = {round(float(fit[0]), 4)}U^2+{round(float(fit[1]), 4)}U+{round(float(fit[2]), 4)}"
@@ -38,24 +31,30 @@ if go == 1:
plt.plot(x, fit_fn(x), color="BLUE", label="T(U)")
plt.scatter(x, y, color="MAGENTA", marker="o", label="Messsdaten")
plt.ylabel("Temperatur")
plt.xlabel("Spannung")
titel = 'Temperatursonde MCP9701A Nummer: {}'.format(n)
plt.title(titel)
plt.axis([0.6, 2, 15, 70])
plt.scatter(x, y, color="MAGENTA", marker="o", label="Data")
plt.ylabel("Temperature")
plt.xlabel("Voltage")
title = 'Sensor MCP9701A #{}'.format(n)
plt.title(title)
plt.axis((0.6, 2.0, 15.0, 70.0))
plt.legend(loc="lower right")
plt.annotate(formula, xy=(0.85, 60))
plt.grid(True)
plt.show()
saveit = input("Soll der Graph gespeichert werden? (y/n) ").lower()
saveit = input("Do you wish to save the plot? (y/N) ").lower()
if saveit == "y":
plt.savefig("Sonde"+str(n)+".png")
plt.savefig("Sonde"+str(n)+".pdf", format="pdf")
plt.savefig("Sonde"+str(n)+".svg", format="svg")
print("saved images")
plt.savefig("Sensor"+str(n)+".png")
plt.savefig("Sensor"+str(n)+".pdf", format="pdf")
plt.savefig("Sensor"+str(n)+".svg", format="svg")
print("==> Images saved")
else:
print("discarded images")
print("==> Images discarded")
filename = input("Please enter a file path to the csv file to be plotted: ")
try:
file = open(filename, "r")
generate_plot()
except FileNotFoundError:
print("Failed to open file (non-existent or corrupted?)")