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

Question

我想计算 A 列和 B 列相交的实例数。A 列和 B 列中的行是字符串列表。例如，A 列可能包含 [car、passenger、truck]，B 列可能包含 [car、house、flower、truck]。由于在这种情况下，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

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

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