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)

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