python - Pandas/Python 相当于 Stata 的 "levelsof"

Question

我想获取数据框中一个变量的所有不同或唯一值的列表，这些值与该数据框中另一个变量的特定值一致。

在 Stata 中我会使用类似的东西:

levelsof(ID1) if ID2==i

我如何在 Python 中执行此操作？

Answer 1

Stata的levelsof相当于pandas的unique()。它们都返回一组唯一或不同的值。

>>> df=pd.DataFrame({ 'id1':[0,0,1,1,2,2], 
                      'id2':[5,5,5,6,6,6] })

   id1  id2
0    0    5
1    0    5
2    1    5
3    1    6
4    2    6
5    2    6

>>> df.loc[ df['id2'] == 5, 'id1' ].unique()

array([0, 1])