使用帶有過濾器的熊貓寫入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 

選擇的df其中mask是真行：

df_filtered = df.loc[mask] 

寫數據幀到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') 

Answer 2

您可以使用panda.DataFrame.query()接口。它允許文本字符串查詢，並且對於大型數據集非常快速。

像這樣的東西應該工作：

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

# the len() method is not available to query, so pre-calculate 
title_len = df["title"].str.len() 

# build the data frame and send to csv file, title_len is a local variable 
df.query('views >= 1000 and starring > 3 and @title_len <= 10').to_csv(...)