python - 按时间范围选择 DataFrame 行

Question

我需要从特定行中选择 +- 1sec 的行(标记为 to_pick)

我可以用循环来做到这一点，但我正在寻找更优雅的方式

import pandas as pd
import numpy as np
N = 1000
tidx = pd.date_range('2019-07-01 09:30:00', periods=N, freq='S')
np.random.seed(3)
data = np.random.randn(N)
ts = tidx + pd.to_timedelta(pd.np.random.randn(N), unit='s')

df = pd.DataFrame({'Time': ts, 'to_pick': pd.np.random.randn(N) > 0.98, 'start': ts - pd.to_timedelta(1, 's'), 'end': ts + pd.to_timedelta(1, 's')})
df.loc[~df['to_pick'], 'start'] = pd.np.nan
df.loc[~df['to_pick'], 'end'] = pd.np.nan

我正在寻找类似的东西

 df['Time'].between(df['start', df['end'])

这将结合条件

Answer 1

如果我理解正确的话，当to_pick = true<时，您希望选择Time在start和end之间的行:

start = df.query('to_pick')['start'].values
end = df.query('to_pick')['end'].values
t = df['Time'].values[:, None]

df.loc[np.any((start <= t) & (t <= end), axis=1)]