pandas 求区间的交集

66 阅读 0 评论 44 点赞

我是靠谱客的博主谦让黑米，最近开发中收集的这篇文章主要介绍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 求区间的交集所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

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