フィルタ付きパンダを使用したcsvへの書き込み

Question

私はPythonを使用してcsvファイルを読み込むために、pandasライブラリを使用しています。

私は、その後のような、特定の値または文の欄をチェックしてい

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

：私は上記の基準を使用して、新しいCSVファイルを作成したいが、それを行う方法がわからないよ

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

これらの属性をすべて1つのcsvに結合する方法について説明します。

誰もが考えている？

Answer 1

&（ビット単位-と）を使用して、ブールのマスクを組み合わせる：

mask = viewNum & starringActorNum & titleLen 

はmaskがTrue dfの行を選択します。

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(...)