python - 如何在 DataFrame 列中匹配 'key off' 对

Question

假设我有一个 df:

src    dest
LV       NC
LA       NY
NC       LV
NY       LA

如何“关闭”唯一的 src/dest 对，以便可以像这样在新列中引用它们？

src    dest     pair
LV       NC       1
LA       NY       2
NC       LV       1
NY       LA       2

df = pd.DataFrame({'src':['LV','LA','NC','NY'], 'dest':['NC', 'NY', 'LV', 'LA']})

Answer 1

尝试使用 tuple + set 然后 factorize

df['pair'] = df.apply(lambda x : tuple(set(x)),1).factorize()[0]+1
Out[69]: array([1, 2, 1, 2], dtype=int64)