通过列表字典过滤a Pandas DataFrame

Question

电影 Dataframe

我有一个包含电影信息的 DataFrame，我正在尝试过滤行，以便如果词典列表包含“名称”==“指定类型”，它将显示包含该类型的电影。

我试过使用列表理解

filter = ['Action']
expectedResult = [d for d in df if d['name'] in filter]

但是我最终遇到了一个错误：

类型错误：字符串索引必须是整数

Answer 1

您的循环for d in df将为您的值提供标题。 您的 d 将具有 generes 作为价值。 尝试运行：-

for d in df:
    print(d)

你会明白的

Answer 2

d 是代码中的列名。 这就是您收到此错误的原因。

请参阅以下示例：

import pandas as pd

df = pd.DataFrame({"abc": [1,2,3], "def": [4,5,6]})
for d in df:
    print(d)

给出：

abc
def

我认为你正在尝试做的事情可以通过以下方式实现：

df = pd.DataFrame({"genre": ["something", "soemthing else"], "abc": ["movie1", "movie2"]})

movies = df.to_dict("records")

[m["abc"] for m in movies if m["genre"] == "something"]

这使：

['movie1']