[英]Pandas select rows when column value within range from another row column value
[英]Pandas select rows when column value within range from another row column value with group filter
我想扩展一个我在问题链接上提出的问题
情况比较复杂,所以我认为那里的解决方案不适合
我正在尝试使用以下格式从数据框(100k-500k行)创建子集
d = {'time':[1,2,3,5,7,9,9.5,10], 'val':['not','match','match','not','not','match','match','match'],
'group':['a','a','b','b','b','a','a','c']}
df = pd.DataFrame(d)
print(df)
group time val
0 a 1.0 not
1 a 2.0 match
2 b 3.0 match
3 b 5.0 not
4 b 7.0 not
5 a 9.0 match
6 a 9.5 match
7 c 10.0 match
当时间在有限范围内时,我想选择一个包含所有行的子集。 例如,如果范围<= 1,则选择第一行和最后三行,它们来自不同的组
而我想要的输出
group time val
1 a 2.0 match
2 b 3.0 match
5 a 9.0 match
6 a 9.5 match
7 c 10.0 match
这适用于您的示例,希望适用于您的数据:
df.loc[((df['time'].diff() <= 1)|(df['time'].diff(-1) >= -1))&((df['group']!=df['group'].shift(-1).fillna(df['group']))|(df['group']!=df['group'].shift(1).fillna(df['group'])))]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.