dev: добавлена возможность обрабатывать трейсы вне зависимости от регистра символов в названии колонок csv файла
This commit is contained in:
parent
b15b079221
commit
8d0151cb97
@ -24,10 +24,34 @@ class DataConverter(BaseDataConverter):
|
|||||||
# TODO: Осмысленное сообщение в лог. Без traceback и прочего подобного.
|
# TODO: Осмысленное сообщение в лог. Без traceback и прочего подобного.
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _fix_headers(dataframe: pd.DataFrame) -> pd.DataFrame:
|
||||||
|
correct_columns = [
|
||||||
|
"time", "Closing", "Electrode Force, N FE", "Electrode Force, N ME",
|
||||||
|
"Force Control FE", "Force Control ME", "Hold Position ME", "Oncomming",
|
||||||
|
"Position Control FE", "Position Control ME", "Relief",
|
||||||
|
"Rotor Position, mm FE", "Rotor Position, mm ME",
|
||||||
|
"Rotor Speed, mm/s FE", "Rotor Speed, mm/s ME",
|
||||||
|
"Squeeze", "Welding", "Welding Current ME", "Welding Voltage ME"
|
||||||
|
]
|
||||||
|
try:
|
||||||
|
correct_mapping = {name.lower(): name for name in correct_columns}
|
||||||
|
new_columns = []
|
||||||
|
for col in dataframe.columns:
|
||||||
|
fixed_col = correct_mapping.get(col.lower(), col)
|
||||||
|
new_columns.append(fixed_col)
|
||||||
|
dataframe.columns = new_columns
|
||||||
|
return dataframe
|
||||||
|
except:
|
||||||
|
# TODO: Обработка исключений!
|
||||||
|
# TODO: Осмысленное сообщение в лог. Без traceback и прочего подобного.
|
||||||
|
return None
|
||||||
|
|
||||||
def convert_data(self, files: list[str]) -> None:
|
def convert_data(self, files: list[str]) -> None:
|
||||||
try:
|
try:
|
||||||
dataframes = [pd.read_csv(file) if file != '' else None for file in files]
|
dataframes = [pd.read_csv(file) if file != '' else None for file in files]
|
||||||
converted_dataframes = list(map(self._replace_bool, dataframes))
|
renamed_dataframes = list(map(self._fix_headers, dataframes))
|
||||||
|
converted_dataframes = list(map(self._replace_bool, renamed_dataframes))
|
||||||
except:
|
except:
|
||||||
# TODO: Обработка исключений!
|
# TODO: Обработка исключений!
|
||||||
# Get the traceback object
|
# Get the traceback object
|
||||||
|
|||||||
7457
trace_samples/point_40000_2025_01_31-14_21_43.csv
Normal file
7457
trace_samples/point_40000_2025_01_31-14_21_43.csv
Normal file
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue
Block a user