社区教程 Wiki

注册登录

创作新主题

社区所有版块导航

Python

python开源 Django Python DjangoApp pycharm

DATA

docker Elasticsearch

问与答闲聊招聘翻译创业分享发现分享创造求职区块链支付之战

aigc

aigc chatgpt

WEB开发

linux MongoDB Redis DATABASE NGINX 其他Web框架 web工具 zookeeper tornado NoSql Bootstrap js peewee Git bottle IE MQ Jquery

机器学习

机器学习算法

Python88.com

反馈公告社区推广

产品

短视频

印度

一周十大热门主题

孙割很会整活！把瓜开源到 GitHub，还发了一个币，就叫“我的女友景甜”

如何让AI从0到1帮忙想深度学习idea（附实操Prompt）

OpenAI高管：ChatGPT与Codex合体，终局是个人AGI

清华这个开源教育Agent夯爆了，Github已经21K Star了～

上海技物所提出超构表面与深度学习协同的高光谱偏振成像方案

OpenAI自研芯片取得突破；来自中国的匿名模型Ox Alpha在海外爆火；腾讯混元Hy4 prev...

〖今日火爆趋势〗GitHub TOP10（2026年08月28日）

整活！孙割把瓜开源到 GitHub + 还有人蹭热点发一个新币，就叫“我的女友景甜”

2026必看AI干货!《大模型/AIGC/GPT-4/Transformer/DL/KG/NLP/C...

单条视频涨粉超18万，《茧物语》拍出了另类超能力女主；AI历史片《问苍生》全网爆火，播放量超5000...

关注

Py学习 » Python

python pandas使用类别标志分组

DBA108642 • 6 年前 • 1758 次点击

我有一个这样的数据框:

|transaction_id|category|
-------------------------
|1234          |Book    |
|1234          |Car     |
|1234          |TV      |
|1235          |Car     |
|1235          |TV      |
|1236          |Car     |

基本上,我想按事务id分组,并创建一个列来标记事务id在category列中是否有相应的tv,因此理想情况下生成的数据帧如下所示:

|transaction_id|HasTV?|
-----------------------
|1234          |Y     |
|1235          |Y     |
|1236          |N     |

我用的是pandas,我知道如何使用groupby函数,我从来没有做过这样的事情,以前有条件检查

Python社区是高质量的Python/Django开发社区
本文地址：http://www.python88.com/topic/41175

文章 [ 1 ] | 最新文章 6 年前

• 1 楼

Randy 6 年前

一种选择是看 .unique() 对于类别,然后对生成的系列进行操作:

In [28]: df.groupby("transaction_id")['category'].unique().apply(lambda x: 'TV' in x)
Out[28]:
transaction_id
1234.0     True
1235.0     True
1236.0    False
Name: category, dtype: bool

另一个可能更快但更模糊的版本是预先测试所需的类别,然后执行groupby:

In [29]: (df['category'] == 'TV').groupby(df["transaction_id"]).max()
Out[29]:
transaction_id
1234.0     True
1235.0     True
1236.0    False
Name: category, dtype: bool

登录后回复

关于移动版

Py学习 - 专注于Python技术发展的社区(原Django社区)