Pandas数据框中列表中的“Where子句”

Question

我有一种叫df的熊猫数据帧：

     email        | list
___________________________
email1@email.com  | [0,1]
email1@email.com  | [2,1]
email1@email.com  | [0,3]
email1@email.com  | [0,0]
email1@email.com  | [0,1]

我想从具有0列表的df中检索所有行：[0,0]

我在做：

df2 = df[df['list'] == [0,0]]

但我得到了以下错误：

ValueError: Arrays were different lengths: 5 vs 2

Answer 1

将列表列表与单个条目进行比较。您应该使用iterrows()来过滤df。iterrows()创建生成元组的生成器，其第二个条目是列字典。您可以遍历它们并与之匹配，然后构建一个新的数据帧。

df2 = {'email':[], 'list':[]}
for row in df.iterrows():
    row_dictionary = row[1]
    if row_dictionary['list'] == [0,0]:
        for key in df2.keys():
            df2[key].append(row_dictionary[key])
df2 = pandas.DataFrame.from_dict(df2)

通过使用字典的键来填充它，您可以在任何数据帧上使用此方法。