dev: добавлена возможность обрабатывать трейсы вне зависимости от регистра символов в названии колонок csv файла
This commit is contained in:
parent
b15b079221
commit
8d0151cb97
@ -23,11 +23,35 @@ class DataConverter(BaseDataConverter):
|
||||
# TODO: Обработка исключений!
|
||||
# TODO: Осмысленное сообщение в лог. Без traceback и прочего подобного.
|
||||
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:
|
||||
try:
|
||||
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:
|
||||
# TODO: Обработка исключений!
|
||||
# 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