使用带过滤器的熊猫写入 csv

Question

我正在使用 Pandas 库使用 Python 加载一个 csv 文件。

import pandas as pd
df = pd.read_csv("movies.csv")

然后我检查特定值或语句的列，例如：

viewNum = df["views"] >= 1000
starringActorNum = df["starring"] > 3
df["title"] = df["title"].astype("str")
titleLen = df["title"].str.len() <= 10

我想使用上述标准创建一个新的 csv 文件，但我不确定如何做到这一点以及如何将所有这些属性组合成一个 csv。

谁有想法？

Answer 1

使用&（按位与）组合布尔掩码：

mask = viewNum & starringActorNum & titleLen

选择dfwheremask为 True的行：

df_filtered = df.loc[mask]

将 DataFrame 写入 csv：

df_filtered.to_csv('movies-filtered.csv')

import pandas as pd

df = pd.read_csv("movies.csv")

viewNum = df["views"] >= 1000
starringActorNum = df["starring"] > 3
df["title"] = df["title"].astype("str")
titleLen = df["title"].str.len() <= 10

mask = viewNum & starringActorNum & titleLen
df_filtered = df.loc[mask]

df_filtered.to_csv('movies-filtered.csv')