Векторизация функции pandas get_lo c - PullRequest
0 голосов
/ 25 апреля 2020

dfMain - это большой фрейм данных, из которого df1 был создан на основе определенных условий. Индекс для dfMain представляет собой уникальную метку времени.

Только для функции pandas .dataframe.index.get_lo c принимает одно значение за раз, можно ли векторизовать следующий код?

    df1 = dfMain[(dfMain['colX'] == 1) &
                            (dfMain['date_minute'].dt.time < pd.Timestamp('10:00').time())]

    # make a list containing the indices
    nextRow = []
    for row in df1.itertuples(index=True):
        nextRow.append(dfMain.index.get_loc(row.Index))

    # get the subsequent row
    nextRow = np.array(nextRow) + 1

Я пытался numpy .where:

np.where(df1['date_minute'] == dfMain['date_minute'])

, но это выдает ошибку Can only compare identically-labeled Series objects

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...