有什么有效的方法可以在 python 中編寫此代碼

Question

我想在 python 中編寫這段代碼。

proc sql;
select count(distinct ID_1)
from DATA
where ID_1 = ID_2 and ID_type in ("11","23","46");
quit;

我可以通過三個步驟做到這一點

a = [x if x==y and z in ("11","23", "46") for x,y,z in zip(DATA['x'],DATA['y'],DATA['z'])]
a = [i for i in a if str(i) != 'nan']
len(np.unique(a))

是否有任何有效的方法來編寫相同的代碼。

Answer 1

最常見的 SQL 操作可以在 python 和 pandas 中輕松轉換：

DATA[(DATA.ID_1 == DATA.ID_2) & (DATA.ID_type.isin(["11", "23", "46"]))].ID_1.nunique()

閱讀pandas 的介紹了解更多信息。

Answer 2

使用query方法進行不同的過濾：

DATA.query('ID_1 == ID_2 and ID_type.isin(["11", "23", "46"])').ID_1.nunique()