当前位置: 代码迷 >> python >> 根据 Pandas 数据框中列中的值过滤数据
  详细解决方案

根据 Pandas 数据框中列中的值过滤数据

热度:113   发布时间:2023-07-14 09:49:55.0

我最近一直在研究一些数据。 在过滤过程中,我发现一些列有一些问题。 我只想在 Branch 列的最后保留那些带有 ')' 的行。

我尝试了几种选择,但我想找到最快的方法来解决它。

由于您没有以文本形式提供数据,我创建了一个示例数据框:

输入:

d = {'college_name': ['College {}'.format(i+1) for i in range(8)], 'branch': ['Civil Enigineering '+ '(4 Years)'*(i%2) for i in range(8)]}
df = pd.DataFrame(data=d, columns=['college_name','branch'])
df

输出:

    college_name    branch
0   College 1   Civil Enigineering
1   College 2   Civil Enigineering (4 Years)
2   College 3   Civil Enigineering
3   College 4   Civil Enigineering (4 Years)
4   College 5   Civil Enigineering
5   College 6   Civil Enigineering (4 Years)
6   College 7   Civil Enigineering
7   College 8   Civil Enigineering (4 Years)

Pandas 系列内置了字符串处理方法。 您可以使用 str.endswith(')') 来过滤您的数据。 请注意df['branch'].str.endswith(')')将返回一个布尔掩码。

输入:

df[df['branch'].str.endswith(')')]

输出:

    college_name    branch
1   College 2   Civil Enigineering (4 Years)
3   College 4   Civil Enigineering (4 Years)
5   College 6   Civil Enigineering (4 Years)
7   College 8   Civil Enigineering (4 Years)
  相关解决方案