主页 / coding / 问题 / 78566699
Accepted
Giampaolo Levorato
Asked: 2024-06-02 22:20:32 +0800 CST

获取 groupby pandas 中的第一个非零元素

  • 772

我创建了以下 pandas 数据框：

import pandas as pd

ds = {'cycle' : [1,1,1,1,2,2,2,2], 'values' : [0,0,-1,1,2,0,0,1]}

df = pd.DataFrame(data=ds)

数据框如下所示：

df

  cycle values
0   1   0
1   1   0
2   1   -1
3   1   1
4   2   2
5   2   0
6   2   0
7   2   1

我需要创建一个新的数据框（称为），其中仅包含第一条记录，每个ds记录的列值均非零。valuescycle

我想这可以通过df.groupby(['cycle'])代码周围的东西来完成。

所以生成的数据框将如下所示：

   cycle    values
0   1        -1
1   2         2
pandas

2 个回答

  1. 删除空项（例如使用query），然后drop_duplicates：

    df.query('values != 0').drop_duplicates(subset='cycle')

    输出：

       cycle  values
2      1      -1
4      2       2
    • 2
  2. Best Answer

    你可以

    • filter出零元素

    • 用于groupby按循环分组

    • 提取first每个组的元素

      df[df['values'] != 0].groupby(['cycle']).first()
    • 1

相关问题