我有以下代碼:
import pandas as pd
df = pd.util.testing.makeDataFrame()
max_index = df.A.idxmax()
我想要做的是max_index在資料框的上方和下方獲取索引值。您能否告知如何實作這一點。
uj5u.com熱心網友回復:
如果您不確定索引是否有重復,一個安全的方法是:
import pandas as pd
df = pd.util.testing.makeDataFrame()
max_index = df.A.idxmax()
before = df['A'].shift(-1).idxmax()
after = df['A'].shift().idxmax()
如果索引是唯一的:
i = df.index.get_loc(max_index)
before, after = df.index[i-1], df.index[i 1]
或者,可能效率稍高一些,并且還可以處理重復的索引:
i = df.reset_index()['A'].idxmax()
before, max_index, after = df.index[i-1:i 2]
轉載請註明出處,本文鏈接:https://www.uj5u.com/ruanti/527147.html
標籤:Python熊猫
