python - 根据列中的最大值过滤数据框- Pandas

使用 Pandas ，我有一个看起来像这样的DataFrame:

Hour            Browser     Metric1   Metric2   Metric3
2013-08-18 00   IE          1000      500       3000
2013-08-19 00   FF          2000      250       6000
2013-08-20 00   Opera       3000      450       9000
2001-03-21 00   Chrome/29   3000      450       9000
2013-08-21 00   Chrome/29   3000      450       9000
2014-01-22 00   Chrome/29   3000      750       9000

我想创建一个浏览器数组，其最大值为Metric1>2000。是否有最佳方法？您基本上可以看到我正在尝试使用以下代码。

browsers = df[df.Metric1.max() > 2000]['Browser'].unique()

最佳答案

您可以对浏览器进行分组并采用最大值:

In [11]: g = df.groupby('Browser')

In [12]: g['Metric1'].max()
Out[12]:
Browser
Chrome/29    3000
FF           2000
IE           1000
Opera        3000
Name: Metric1, dtype: int64

In [13]: over2000 = g['Metric1'].max() > 2000

In [14]: over2000
Out[14]:
Browser
Chrome/29     True
FF           False
IE           False
Opera         True
Name: Metric1, dtype: bool

要获取数组，请将其用作 bool 掩码:

In [15]: over2000[over2000].index.values
Out[15]: array(['Chrome/29', 'Opera'], dtype=object)

关于python - 根据列中的最大值过滤数据框- Pandas ，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/18906530/