python - Pandas 选择符合我标准的特定行之前和之后的特定行

Question

我有一个数据框。第一行的任何地方都有一个唯一的值可以满足我的标准。例如 t = 800

现在我想找到该值所在的行，并从它之前的 n 行到之后的 n 行中进行选择。

我试过:

idx = df[df['t']==stime]
from_idx = idx-1200
to_idx = idx + 1200
vib = df.iloc[to_idx:from_idx]

它找到了符合我的标准的行，但我无法从我的数据框中得到正确的选择

Answer 1

idx 在您的逻辑中是一个 pd.DataFrame 对象，而不是表示行索引的标量。如果您提取标量索引，假设您的索引是唯一的，您可以使用 pd.Index.get_loc获取整数位置:

idx = df.index.get_loc(df[df['t'] == stime].index[0])
vib = df.iloc[idx - 1200: idx + 1200]