fix: Исправил обработку начала со статусом "Oncoming"

This commit is contained in:
Леонид Титов 2024-12-05 13:27:54 +03:00
parent 88003f4f02
commit 6a244afadd

View File

@ -365,9 +365,11 @@ class BasePointPassportFormer:
times:pd.Series, times:pd.Series,
dataframe: pd.DataFrame) -> list[list[float]]: dataframe: pd.DataFrame) -> list[list[float]]:
start_idx, finish_idx = self._find_indexes(signal, dataframe) start_idx, finish_idx = self._find_indexes(signal, dataframe)
if start_idx[0] > finish_idx[0]:
start_idx = np.insert(start_idx, 0, 0)
print (start_idx)
start_list = times.loc[start_idx].tolist() start_list = times.loc[start_idx].tolist()
end_list = times.loc[finish_idx].tolist() end_list = times.loc[finish_idx].tolist()
#if start_idx[0] > finish_idx[0]:
if len(start_list) - len(end_list) == 1: if len(start_list) - len(end_list) == 1:
end_list.append(float(times[len(times)-1])) end_list.append(float(times[len(times)-1]))
return start_list, end_list return start_list, end_list