Pandasデータフレームで条件により行を絞り、新たなデータフレームにする。

 

 

#sample_pandas_normal.csv on desktop

name,age,state,point
Alice,24,NY,64
Bob,42,CA,92
Charlie,18,CA,70
Dave,68,TX,70
Ellen,24,CA,88
Frank,30,NY,57

 

#code

import pandas as pd

print(pd.__version__)

# 2.0.3

path_file="C:/Users/shienkikou11/Desktop/sample_pandas_normal.csv"

df = pd.read_csv(path_file)

print(df)

val = 25

state = "CA"

df2=df.query('state == @state')

print(df2,"\n",type(df2),"\n num_row_df2 :",len(df2),"\n num_column_df2 :",len(df2.columns))

 

#result

1.4.3
      name  age state  point
0    Alice   24    NY     64
1      Bob   42    CA     92
2  Charlie   18    CA     70
3     Dave   68    TX     70
4    Ellen   24    CA     88
5    Frank   30    NY     57
      name  age state  point
1      Bob   42    CA     92
2  Charlie   18    CA     70
4    Ellen   24    CA     88 
 <class 'pandas.core.frame.DataFrame'> 
 df2 num_row : 3 
 df2 num_column : 4