You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何用代码返回Value列含负数的Name对应的所有行?

解决方案:筛选存在负数的分组所有行

这是个典型的分组后条件筛选需求,用Python的pandas库可以快速实现,我给你一步步说明:

1. 先构建示例数据集

首先把你提供的原始数据转换成pandas的DataFrame格式:

import pandas as pd

data = {
    'Name': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'C', 'D', 'D'],
    'Date': ['2008-04', '2008-06', '2009-05', '2010-04', '2005-04', '2006-05', '2010-08', '2012-05', '2016-05', '2017-05'],
    'Value': [-0.5, 3, -0.8, -0.6, 2, 2, -1, 2, 5, 6]
}
df = pd.DataFrame(data)

2. 方法一:先筛选目标分组,再提取数据

先找出所有至少有一个负数Value的Name,再用这些Name过滤原数据集:

# 按Name分组,检查每个组是否存在负数,提取符合条件的Name
target_names = df.groupby('Name')['Value'].apply(lambda group: (group < 0).any())
target_names = target_names[target_names].index

# 筛选出这些Name的所有行
result = df[df['Name'].isin(target_names)]

3. 方法二:用transform简化写法(更高效)

transform方法可以直接返回和原数据行对齐的布尔值,一步完成筛选,代码更简洁:

# 对每个分组检查是否存在负数,用结果直接过滤原数据
result = df[df.groupby('Name')['Value'].transform(lambda x: (x < 0).any())]

验证结果

运行后result会包含A的全部4行和B的全部3行,自动排除C、D的行,输出如下:

Name     Date  Value
0    A  2008-04   -0.5
1    A  2008-06    3.0
2    A  2009-05   -0.8
3    A  2010-04   -0.6
4    B  2005-04    2.0
5    B  2006-05    2.0
6    B  2010-08   -1.0

内容的提问来源于stack exchange,提问作者Peter

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.05.28 06:25:25