Python：如何根据每个条目中的行数删除dict条目

Question

dict1 = {k: v for k, v in df.groupby('Some Column Name')}

dict2 = {k: v for k, v in dict1.items() if v[0] <=20}

Answer 1

您可以使用df.shape[0]并使用您已经编写的代码。df.shape中的第一个值是行数，第二个值是列数

dict2 = {k: v for k, v in dict1.items() if df.shape[0]<21}

我创建了这个小数据帧来向您展示：

数据帧为：

df = pd.DataFrame([['ch',2],['bt',4],['ch',10],['bt',5],['aw',3]],columns=['code','freq'])
print(df.shape)
(5,2)  #5 rows is the upper limit

dict1 = {k: v for k, v in df.groupby('code') if df.shape[0]>2}  
print(dict1)

{'aw':   code  freq
4   aw     3, 'bt':   code  freq
1   bt     4
3   bt     5, 'ch':   code  freq
0   ch     2
2   ch    10}

dict1 = {k: v for k, v in df.groupby('code') if df.shape[0]>6}
print(dict1)
{}