パンダ: query() 関数で変数を使用する方法
次の構文を使用して、pandas でquery()関数を使用し、変数名を参照できます。
df. query (' team == @team_name ')
この特定のクエリは、pandas DataFrame 内で、チーム列がteam_nameという変数に格納されている値と等しい行を検索します。
次の例は、この構文を実際に使用する方法を示しています。
例: Pandas クエリで変数を使用する方法
さまざまなバスケットボール選手に関する情報を含む次のパンダ データフレームがあるとします。
import pandas as pd #createDataFrame df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'B', 'B', 'B', 'B', 'C', 'C', 'C', 'C '], ' position ':['G', 'G', 'F', 'G', 'F', 'F', 'F', 'G', 'G', 'F', 'F'], ' points ': [22, 25, 24, 39, 34, 20, 18, 17, 20, 19, 22]}) #view DataFrame print (df) team position points 0 AG 22 1 AG 25 2 AF 24 3 BG 39 4 BF 34 5 BF 20 6 BF 18 7 GC 17 8 GC 20 9 CF 19 10 CF 22
ここで、チーム列の値が C に等しい行を検索するとします。
次の構文を使用して、「C」に等しいteam_nameという変数を作成し、この変数をquery()関数で参照できます。
#specify team name to search for team_name = ' C ' #query for rows where team is equal to team_name df. query (' team == @team_name ') team position points 7 C G 17 8 C G 20 9 C F 19 10 C F 22
query()関数は、チーム列の値が C に等しいすべての行を返すことに注意してください。
また、必要に応じて、 query()関数で複数の変数を参照できることにも注意してください。
たとえば、次のコードは、 query()関数を使用して、 team列の値がteam_Aという変数またはteam_Cという変数の値と等しいすべての行を返す方法を示しています。
#create two variables team_A = ' A ' team_C = ' C ' #query for rows where team is equal to either of the two variables df. query (' team == @team_A | team == @team_C ') team position points 0 A G 22 1 A G 25 2 A F 24 7 C G 17 8 C G 20 9 C F 19 10 C F 22
クエリは、 teamが指定した 2 つの変数のいずれかに格納されている値と等しい DataFrame 内のすべての行を返します。
注: pandas query()関数の完全なドキュメントはここで見つけることができます。
追加リソース
次のチュートリアルでは、パンダで他の一般的なタスクを実行する方法を説明します。
Pandas: 文字列の長さに基づいて行をフィルタリングする方法
Pandas: 条件に基づいて行を削除する方法
パンダ: 「NO IN」フィルターの使用方法