У меня есть огромный CSV-файл, который содержит данные о вождении велосипеда.
Итак, я получил столбец времени в секундах и столбец скорости.
Я хотел бы проверить конкретный шаблон в данных, чтобы определить, что произошло на дороге.
Например, вождение на светофоре:
Я получил это до сих пор:
import pandas as pd
df = pd.read_csv('.csv', usecols = ['time', 'speed'])
df['accelerating'] = df['speed'].diff() > 0
И я хочу что-то вроде этого:
df_traffic_light = df.loc[df['speed'] < 15 & accelerating == False #driving torwards the traffic light;
& df.loc[df['speed']< 1 #getting really slow or Standing still;
& df.loc[df['speed'] > 5 & accelerating == True #for light switched to green and starting again
Ожидаемый результат:
time speed acceleration
0 5.000 14.0 false
1 7.056 12.0 false
2 10.097 8.0 false
3 12.131 1.0 false
4 14.165 0.0 false
5 16.201 0.0 false
6 18.236 2.0 true
7 20.267 4.0 true
Я попробовал это с dataframe.rolling, но не сработало. Есть идеи, как мне это решить?