Files
AERO4450-Graphs/main.py
2023-04-02 20:08:51 +10:00

39 lines
1.4 KiB
Python

# ENGG4900 Graph Generators
# Cal Wing 2023
import numpy as np
import pandas as pd
from makeGraph import makeGraph, UQ_COLOURS as UQC
DATA_FILE_PATH = ".\\data\\AERO4450-Group-10-Data-Analysis-v2.xls"
actualData = pd.read_excel(DATA_FILE_PATH, sheet_name="Actual Data")
calibrationData = pd.read_excel(DATA_FILE_PATH, sheet_name="Calibration Data Graphs")
def main():
testRun = 0
cols = [f"Unnamed: {3 + testRun*5}", f"Unnamed: {4+ testRun*5}"]
data = actualData[cols].rename(columns = {cols[0]:'Time', cols[1]:'Strain'})
data = data[5:505].astype({"Time":"int", "Strain":"float"})
data1 = data[0::2]
data2 = data[1::2]
makeGraph({
"title": f"Strain over Time\nRun: {testRun+1}",
"xLabel": "Time (ms) - Δt = 10ms",
"yLabel": "Strain (ε)",
"plots":[
{"type":"scatter", "x": data1["Time"], "y": data1["Strain"], "args":{"s":10, "zorder":2}, "label":"Upper", "colour": UQC["purple"]},
{"type":"scatter", "x": data2["Time"], "y": data2["Strain"], "args":{"s":10, "zorder":1}, "label":"Lower", "colour": UQC["dark_grey"]},
{"type":"plot", "x": data1["Time"], "y": data1["Strain"], "args":{"zorder":0, "alpha":0.25}, "label":"Upper", "colour": UQC["purple"]},
{"type":"plot", "x": data2["Time"], "y": data2["Strain"], "args":{"zorder":0, "alpha":0.25}, "label":"Lower", "colour": UQC["dark_grey"]},
]
})
if __name__ == "__main__":
main()