根据两个条件从数据框中删除行。

Question

根据两个条件从数据框中删除行。

3

我有一个 Python 数据框，其中包含多个赛季的 NHL 球员数据。我想要删除所有未参加 2018-2019 赛季的球员行。例如，如果 Joe Jones 在 2018-2019 赛季中打了球，我希望保留他在那个赛季的数据以及他参加的其他任何赛季的数据。

我考虑代码可能是这样的：

for player in data.players:
      if data['Year'] == '2018-2019':
          save player's name
      else:
          remove player's data

例如，我的数据框长这样：

Year         Player     TM     GP
2018-2019    Joe        MTL    78
2017-2018    Joe        MTL    82
2016-2017    Joe        MTL    80
2017-2018    Jim        STL    76
2016-2017    Jim        STL    82
2018-2019    Jack       MIN    82

结果将是:

Year         Player     TM     GP
2018-2019    Joe        MTL    78
2017-2018    Joe        MTL    82
2016-2017    Joe        MTL    80 
2018-2019    Jack       MIN    82

- SD_23

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- piRSquared · Accepted Answer

`groupby.filter`

df.groupby('Player').filter(lambda d: '2018-2019' in {*d.Year})

        Year Player   TM  GP
0  2018-2019    Joe  MTL  78
1  2017-2018    Joe  MTL  82
2  2016-2017    Joe  MTL  80
5  2018-2019   Jack  MIN  82

使用values数组替代set。

df.groupby('Player').filter(lambda d: '2018-2019' in d.Year.values)

使用NumPy的过度设计

m = df.Year.values == '2018-2019'
i, u = pd.factorize(df.Player)
a = np.zeros(len(u), bool)
np.logical_or.at(a, i, m)

df[a[i]]

        Year Player   TM  GP
0  2018-2019    Joe  MTL  78
1  2017-2018    Joe  MTL  82
2  2016-2017    Joe  MTL  80
5  2018-2019   Jack  MIN  82