如何从 python pandas 中的两个系列中获取元素明智的交集

Question

我的问题是 python pandas。我有两个系列，每个系列都有如下字符串元素：为了简化，我在 DataFrame 中连接了两个系列。

import pandas as pd
import numpy as np
my_df = pd.DataFrame([['ab', 'bz', 'b'], ['cd', 'ct', 'c'], ['ef', 'ka', np.nan]], columns=['sr_1', 'sr_2', 'intersection'])

有什么想法吗？

Answer 1

这是你可以做的：

import pandas as pd
import numpy as np

df1 = pd.DataFrame({'sr1' : ['ab','cd','ef'] ,
                    'sr2' : ['bz','ct','ka',]})

df1['intersection'] = df1.apply(lambda x: set(x.sr1) & set(x.sr2), axis=1)

df1['intersection'] = df1.intersection.apply(lambda x: list(x)[0] if len(x)>0 else np.nan)

输出：

如何从 python pandas 中的两个系列中获取元素明智的交集

How to get element wise intersection from two Series in python pandas

intersection

series

elementwise-operations