（pandas）在顺序无关紧要的情况下，如何基于三列相似的数据创建唯一标识符？

Question

（Python/熊猫） 我正在对英国房价数据做一些分析，看房价是否与附近学校的质量有关。我已将最近三所学校的URN（唯一参考号）与每笔房价交易进行了匹配。数据中的URN\ 1、URN\ 2、URN\ 3列

我想在数据上估计一个固定效应模型，其中固定效应基于最近的三所学校。因此，我想为三所学校的每个集群创建一个唯一的ID，并且我希望这不受学校顺序的影响。房产A和房产B应该有相同的ID，尽管学校的顺序不同

Property    URN_1   URN_2   URN_3
A         100053   100052   100054
B         100052   100054   100053

有人知道我如何使用Python创建唯一的集群id吗

我尝试使用.groupby（）用下面的代码创建ID，但是当学校的顺序不同时，这会给出不同的集群ID

以下是我尝试过的：

import pandas as pd
URN1=[1,2,3,4,5]
URN2=[5,4,3,2,1]
URN3=[1,2,3,2,1]
lst=['a','b','c','d','e']
df=pd.DataFrame(list(zip(URN1,URN2,URN3)),
columns['URN_1','URN_2','URN_3'],index=lst)
df['clusterid']=df.groupby(['URN_1','URN_2','URN_3']).ngroup()
print(df)

我想让观测值'a'和'e'具有相同的集群id，但是通过这种方法给它们提供了不同的id

Answer 1

您可以使用已排序的3个骨灰盒为每个骨灰盒创建一个字符串。
然后按这个新变量分组，并像以前一样使用ngroup（）

df['URN_join'] = df[['URN_1','URN_2','URN_3']].apply(lambda x: '_'.join([str(nb) for nb in sorted(x)]), axis=1)
df['clusterid'] = df.groupby(['URN_join']).ngroup()
df

输出：

    URN_1   URN_2   URN_3   clusterid   URN_join
a   1       5       1       0           1_1_5
b   2       4       2       1           2_2_4
c   3       3       3       2           3_3_3
d   4       2       2       1           2_2_4
e   5       1       1       0           1_1_5