pandas 求区间的交集

233 阅读 0 评论 154 点赞

我是靠谱客的博主谦让黑米，这篇文章主要介绍pandas 求区间的交集，现在分享给大家，希望可以做个参考。

import pandas as pd
import numpy as np

df=pd.DataFrame({'left':[12,5,19],'right':[19,18,20]},index=[*'abc'])
df

如上图，a=[12,19],b=[5,18],c=[19,20]，需求为求两两之间的交集范围，返回交集的区间。

分析，先将区间表示出来，实现上可以用interval，不过更方便的是借助portion库：pip install portion。

import portion as p


df.loc[:,'区间'] = df.apply(lambda x: p.closed(x.left,x.right),axis=1)

co_dic={}
for i in range(len(df.区间)):
    aus=df.区间.apply(lambda x: x & df['区间'][i])
    co_dic[df.区间.index[i]]=aus

pd.DataFrame(co_dic)

pd.concat([df,pd.DataFrame(co_dic)],axis=1)

这样就完成了实现了需求。

以上就是谦让黑米最近收集整理的关于pandas 求区间的交集的全部内容，更多相关pandas内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。