熊貓時間序列條件滿足顯示接下來的幾行

Question

我有一個熊貓數據框加載了來自Yahoo Finance的AAPL的日期數據。我正在尋找數據中遇到的某個條件（例如，當一天接近一天的最低價）和之後（例如，在2012-01-01滿足條件），我想要立即看到接下來的幾行（我對2012-01-02和2012-01-03等AAPL的收盤價格感興趣）。

我使用了數據幀查詢，它只導致條件匹配的行。我不知道如何在匹配的條件行之後立即提取行。

我該如何去做到這一點？ 「查詢」是錯誤的方法嗎？

謝謝 d

Answer 1

你知道你將有一個確切的結果呢？然後，你可以讓他們更多或更少的手工如下：

In [44]: df # Example DataFrame 
Out[44]: 
    c1 c2 
0 a 5 
1 b 10 
2 c 15 

In [45]: index = df[df['c1'] == 'b'].index[0] # Get the index of the row of interest. 

In [47]: df.iloc[range(index, index + 2)] # Get that row and whatever follows immediately after 
Out[47]: 
    c1 c2 
1 b 10 
2 c 15 

如果你有一個以上的結果，遍歷df[df['c1'] == 'b'].index代替。