python - 获取数据框中字典的长度

Question

我目前正在学习 pandas，想知道如何过滤列(即字典)中包含超过 3 个键的行。例如，

data = {'id':[1,2,3], 'types': [{1: 'a', 2:'b', 3:'c'},{1: 'a', 2:'b', 3:'c', 4:'d'}, {1: 'a', 2:'b', 3:'c'}]}
df = pd.dataframe(data)

如何获取列类型中字典长度> 3的行我尝试做

df[len(df['types']) > 3]

但它不起作用。有什么简单的解决方案吗？

Answer 1

使用Series.apply或Series.map :

df = df[df['types'].apply(len) > 3]
#alternative
#df = df[df['types'].map(len) > 3]
print (df)
   id                             types
1   2  {1: 'a', 2: 'b', 3: 'c', 4: 'd'}

或者Series.str.len :

df = df[df['types'].str.len() > 3]