python - 获取 pandas 数据框中出现特定值的第一个月和最后一个月

Question

我有以下内容。数据框:

datetime    Year    cal val
1/1/2000    2000    1   0.23
1/2/2000    2000    1   0.34
2/3/2000    2000    1   0.45
4/4/2000    2000    2   0.56
4/5/2000    2000    2   0.67
4/6/2000    2000    2   0.78
5/7/2000    2000    3   0.89
7/8/2000    2000    3   1
7/9/2000    2000    3   1.11

数据时间列表示数据帧索引。

我想找到最后一个 cal == 1 值的月份(这将是二月)。同样，我想找到第一个 cal == 3 值的月份(在本例中为 may)。现在，我可以做子集:

df[df.cal == 1]

但不确定如何继续。

Answer 1

使用 loc 查找与条件匹配的子集，然后使用 [0] 访问结果索引的第一个元素，使用 [ 访问最后一个元素-1]。使用.month获取相关月份的整数。

“我想找到最后一个 cal == 1 值的月份(这将是二月)。”

>>> df.loc[df.cal == 1].index[-1].month
2

“同样，我想找到第一个 cal == 3 值的月份(在本例中为 may。)。”

>>> df.loc[df.cal == 3].index[0].month
5