从 Pandas Dataframe 列返回最大值和最小值的实际索引值

Question

如何打印/返回特定值的索引？

movies_per_year = pd.DataFrame(movies_per_year)
movies_per_year.columns = ["count"]
print(movies_per_year)

        count
year       
1891      1
1893      1
1894      2
1895      2
1896      2
1898      5

在这里，我的索引是年份。我想返回 count 为 1 的所有索引。此外，movies_per_year.max() 返回 5。因此，它应该返回 1898。

Answer 1

np.where() - 返回满足给定条件的元素的位置索引：

In [31]: np.where(df['count'] == 1)[0]
Out[31]: array([0, 1], dtype=int64)

或

In [35]: np.nonzero(df['count'] == 1)
Out[35]: (array([0, 1], dtype=int64),)

如果您需要真正的索引值（标签）而不是它们的位置：

In [40]: df.index[df['count'] == 1]
Out[40]: Int64Index([1891, 1893], dtype='int64', name='year')

查找最大元素的索引：

In [32]: df['count'].idxmax()
Out[32]: 1898