按列表长度过滤

Question

给出以下代码，我希望获得包含license_no的行和包含两个项目的列表。我应该如何表述过滤器语句?

import pandas as pd
df = pd.DataFrame({'id': ['1','2'],'license_no':[['John','John'],['Mary','Mary','Mary']]})

Answer 1

您可以使用apply函数来获取每行列表的长度，然后使用结果来过滤数据框。

df[df["license_no"].apply(len) == 2]