在Python Pandas中查找兩列的交集 - >字符串列表

Question

我想計算A列和B列相交的實例數。 A列和B列中的行是字符串列表。 例如，列A可以包含[汽車，乘客，卡車]，列B可以包含[汽車，房屋，花卉，卡車]。 由於在這種情況下，2個字符串重疊，列C應顯示 - > 2

我試過了（這些都沒有）：

df['unique'] = np.unique(frame[['colA', 'colB']])

要么

def unique(colA, colB):
    unique1 = list(set(colA) & set(colB))
    return unique1

df['unique'] = df.apply(unique, args=(df['colA'], frame['colB']))

TypeError :('unique（）需要2個位置參數，但3個被賦予'，'發生在索引文章'）

Answer 1

我相信在列表理解中需要set.intersection length ：

df['C'] = [len(set(a).intersection(b)) for a, b in zip(df.A, df.B)]

要么：

df['C'] = [len(set(a) & set(b)) for a, b in zip(df.A, df.B)]

樣品 ：

df = pd.DataFrame(data={'A':[['car', 'passenger', 'truck'], ['car', 'truck']],
                        'B':[['car', 'house', 'flower', 'truck'], ['car', 'house']]})
print (df)
                         A                            B
0  [car, passenger, truck]  [car, house, flower, truck]
1             [car, truck]                 [car, house]

df['C'] = [len(set(a).intersection(b)) for a, b in zip(df.A, df.B)]
print (df)
                         A                            B  C
0  [car, passenger, truck]  [car, house, flower, truck]  2
1             [car, truck]                 [car, house]  1