我在 Python 中有以下 Pandas 資料框:
import pandas as pd
d = {'col1': [1, 2, 3, 4, 5], 'col2': [6, 7, 8, 9, 10]}
df = pd.DataFrame(data=d)
df.index=['A', 'B', 'C', 'D', 'E']
df
這給出了以下輸出:
col1 col2
A 1 6
B 2 7
C 3 8
D 4 9
E 5 10
我需要撰寫一個函式(假設名稱為getNrRows(fromIndex)),它將索引值作為輸入,并回傳給定索引和資料幀的最后一個索引之間的行數。
例如:
nrRows = getNrRows("C")
print(nrRows)
> 2
因為從索引 C 到索引 E 需要 2 個步驟(行)。
如何以最優雅的方式撰寫這樣的函式?
uj5u.com熱心網友回復:
為了您的資訊,我們有內置功能 get_indexer_for
len(df)-df.index.get_indexer_for(['C'])-1
Out[179]: array([2], dtype=int64)
uj5u.com熱心網友回復:
最簡單的方法可能是
len(df[row_index:]) - 1
轉載請註明出處,本文鏈接:https://www.uj5u.com/ruanti/333453.html
