Skip to content

Commit da2df36

Browse files
committed
Commit jumenberger#2 Add method to Mission class to extract csv data
1 parent 8c5b71e commit da2df36

1 file changed

Lines changed: 11 additions & 3 deletions

File tree

uuv_mission/dynamic.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from dataclasses import dataclass
33
import numpy as np
44
import matplotlib.pyplot as plt
5+
import pandas as pd
56
from .terrain import generate_reference_and_limits
67

78
class Submarine:
@@ -74,9 +75,16 @@ def random_mission(cls, duration: int, scale: float):
7475
return cls(reference, cave_height, cave_depth)
7576

7677
@classmethod
77-
def from_csv(cls, file_name: str):
78-
# You are required to implement this method
79-
pass
78+
def from_csv(cls, file_name: str): # creates mission instance by extracting data from a CSV file
79+
mission_data = pd.read_csv(file_name)
80+
81+
reference = mission_data['reference'].to_numpy()
82+
cave_height = mission_data['cave_height'].to_numpy()
83+
cave_depth = mission_data['cave_depth'].to_numpy()
84+
85+
return cls(reference, cave_height, cave_depth)
86+
87+
8088

8189

8290
class ClosedLoop:

0 commit comments

Comments
 (0)