|
4 | 4 | __license__ = "MIT"
|
5 | 5 |
|
6 | 6 | import numpy as np
|
| 7 | +import pandas as pd |
7 | 8 | from scipy.integrate import solve_ivp
|
8 | 9 |
|
9 | 10 | from .Environment import Environment
|
@@ -197,3 +198,64 @@ def du(z, u):
|
197 | 198 | velocityFunction()
|
198 | 199 |
|
199 | 200 | return altitudeFunction, velocityFunction, final_sol
|
| 201 | + |
| 202 | +def create_dispersion_dictionary(dic): |
| 203 | + dataframe = pd.read_csv(dic,sep = ';', skiprows=[0,1], header = None) |
| 204 | + |
| 205 | + |
| 206 | + rocketKeys = list(dataframe[1].dropna()) |
| 207 | + rocketValues = list(dataframe[2].dropna()) |
| 208 | + rocketSD = list(dataframe[3]) |
| 209 | + |
| 210 | + motorKeys = list(dataframe[7].dropna()) |
| 211 | + motorValues = list(dataframe[8].dropna()) |
| 212 | + motorSD = list(dataframe[9]) |
| 213 | + |
| 214 | + launchKeys = list(dataframe[13].dropna()) |
| 215 | + launchValues = list(dataframe[14].dropna()) |
| 216 | + launchSD = list(dataframe[15]) |
| 217 | + |
| 218 | + parachuteKeys = list(dataframe[19].dropna()) |
| 219 | + parachuteValues = list(dataframe[20].dropna()) |
| 220 | + parachuteSD = list(dataframe[21]) |
| 221 | + |
| 222 | + |
| 223 | + allValues = [] |
| 224 | + # crating the dictionary |
| 225 | + |
| 226 | + for i in range(0,len(rocketKeys)): |
| 227 | + |
| 228 | + if pd.isnull(rocketSD[i]): |
| 229 | + allValues.append(rocketValues[i]) |
| 230 | + else: |
| 231 | + allValues.append(((rocketValues[i]), (rocketSD[i]))) |
| 232 | + |
| 233 | + for j in range(0,len(motorKeys)): |
| 234 | + |
| 235 | + if pd.isnull(motorSD[j]): |
| 236 | + allValues.append(motorValues[j]) |
| 237 | + else: |
| 238 | + allValues.append(((motorValues[j]), (motorSD[j]))) |
| 239 | + |
| 240 | + for k in range(0,len(parachuteKeys)): |
| 241 | + |
| 242 | + if pd.isnull(parachuteSD[k]): |
| 243 | + allValues.append(parachuteValues[k]) |
| 244 | + else: |
| 245 | + allValues.append(((parachuteValues[k]), (parachuteSD[k]))) |
| 246 | + |
| 247 | + for l in range(0,len(launchKeys)): |
| 248 | + |
| 249 | + if pd.isnull(launchSD[l]): |
| 250 | + allValues.append(launchValues[l]) |
| 251 | + else: |
| 252 | + allValues.append(((launchValues[l]), (launchSD[l]))) |
| 253 | + |
| 254 | + |
| 255 | + allKeys = rocketKeys + motorKeys + parachuteKeys + launchKeys |
| 256 | + |
| 257 | + analysis_parameters = dict(zip(allKeys,allValues)) |
| 258 | + return analysis_parameters |
| 259 | + |
| 260 | + |
| 261 | + |
0 commit comments