從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