Make graphs
This commit is contained in:
28
main.py
28
main.py
@@ -3,13 +3,37 @@
|
||||
|
||||
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"
|
||||
DATA_FILE = pd.read_excel(DATA_FILE_PATH)
|
||||
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():
|
||||
print(DATA_FILE)
|
||||
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()
|
||||
Reference in New Issue
Block a user