獲取dataframe中某列第一次出現的索引

Question

我有以下 dataframe：

import pandas as pd
d = {'Stages': ['Stage 1', 'Stage 2', 'Stage 2', 'Stage 2', 'Stage 3', 'Stage 1'], 'Start(s)': [0, 630, 780, 840, 900, 930], 'End(s)': [630, 780, 840, 900, 930, 960]}
df = pd.DataFrame(data=d)

    Stages         Start(s) End(s)
0   Stage 1          0      630
1   Stage 2         630     780
2   Stage 2         780     840
3   Stage 2         840     900
4   Stage 3         900     930
5   Stage 1         930     960

我想獲取第 2 階段首次出現在“階段”列中的索引。在此示例中，它將是 1。

我嘗試閱讀有關類似問題的討論，但無法實施。

Answer 1

如果始終存在至少一個Stage 2 ，請使用Series.idxmax比較第一個值：

print (df['Stages'].eq('Stage 2').idxmax())
1

如果可能不存在像Stage 8這樣的next和iter技巧：

print (next(iter(df.index[df['Stages'].eq('Stage 8')]), 'not exist'))
not exist

print (next(iter(df.index[df['Stages'].eq('Stage 2')]), 'not exist'))
1

因為如果不存在匹配值idxmax返回第一個False值：

print (df['Stages'].eq('Stage 8').idxmax())
0

另一個想法是通過Series.where和Series.first_valid_index測試不缺失值的第一個索引：

print (df['Stages'].where(df['Stages'].eq('Stage 8')).first_valid_index())
None

print (df['Stages'].where(df['Stages'].eq('Stage 2')).first_valid_index())
1