pandas 获取每组具有最小列值的行

Question

我想获取每组具有最小列值的所有行，

示例：

df = pd.DataFrame({'asset_symbol': ['100', '100', '100', '1015', '1015'],
 'percent_thresh': [0.75, 0.85, 0.95, 0.75, 0.85],
 'rank': [7.0, 7.0, 4.0, 2.0, 3.0]})

+--------------+----------------+------+
| asset_symbol | percent_thresh | rank |
+--------------+----------------+------+
| 100          | 0.75           | 7    |
+--------------+----------------+------+
| 100          | 0.85           | 7    |
+--------------+----------------+------+
| 100          | 0.95           | 4    |
+--------------+----------------+------+
| 1015         | 0.75           | 2    |
+--------------+----------------+------+
| 1015         | 0.85           | 3    |
+--------------+----------------+------+

想要table：

+--------------+----------------+------+
| asset_symbol | percent_thresh | rank |
+--------------+----------------+------+
| 100          | 0.95           | 4    |
+--------------+----------------+------+
| 1015         | 0.75           | 2    |
+--------------+----------------+------+

我的尝试是：

def max_row(df, column):
    return df.loc[df[column].idxmin()]

df.groupby('asset_symbol').apply(max_row, 'rank')

不过我通常不会使用 apply

Answer 1

IIUC,

df.loc[df.groupby('asset_symbol')['rank'].idxmin()]

输出：

asset_symbol  percent_thresh  rank
2          100            0.95   4.0
3         1015            0.75   2.0

Answer 2

让我们做 sort_values + drop_duplicates

df.sort_values('rank').drop_duplicates('asset_symbol')
  asset_symbol  percent_thresh  rank
3         1015            0.75   2.0
2          100            0.95   4.0

pandas 获取每组具有最小列值的行

pandas get row with min column value per group

python

group-by

rank

pandas