如何“关闭”DataFrame 列中的匹配对

Question

在你的test_foo.py：import unittestfrom unittest import mockfrom foo import Beef, Fooclass TestBeef(unittest.TestCase):      def test_run(self):        mock.object.patch(Foo, 'bar', return_value={'a': 'x'})        beef = Beef()        assert result == {'a': 'x'}

Answer 1

然后尝试使用tuple+setfactorize

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

Answer 2

您可以对数据进行排序并使用groupby().ngroup()：

df['pair'] = (pd.DataFrame(np.sort(df, axis=1))

   .groupby([0,1], sort=False).ngroup() + 1

)

输出：

  src dest  pair

0  LV   NC     1

1  LA   NY     2

2  NC   LV     1

3  NY   LA     2

Answer 3

您可以按以下方式分组frozenset：

df['pair'] = df.groupby(df.apply(frozenset, axis=1)).ngroup() + 1

print(df)

印刷：

  src dest  pair

0  LV   NC     1

1  LA   NY     2

2  NC   LV     1

3  NY   LA     2