如何基于两列过滤时间序列熊猫数据帧?

问题描述 投票:0回答:1

因此,我有一个时间序列数据框,其中一栏既有患者ID,又一栏有年(1-3)。我想使用熊猫来做的是,对于每个唯一的患者ID,只保留所有1-3年级都有条目的行。

IN

ID  Year 
111  1 
111  2 
111  3 
222  1 
222  2
333  1
333  2
333  3

OUT

ID  Year 
111  1 
111  2 
111  3 
333  1
333  2
333  3
python pandas data-science
1个回答
0
投票

可以像这样在一行中完成:

df = pd.DataFrame({
    "ID": [111, 111, 111, 222, 222, 333, 333, 333],
    "Year": [1, 2, 3, 1, 2, 1, 2, 3]})

# filter unique IDs that doesn't contain [1, 2, 3]
df = df.groupby('ID').filter(lambda g: (len(set(g["Year"].values)) == 3)) 
print(df)
#    ID  Year
#0  111     1
#1  111     2
#2  111     3
#5  333     1
#6  333     2
#7  333     3
© www.soinside.com 2019 - 2024. All rights reserved.