【Python,pandas】データフレームの複数のカラムから同じ値を抽出したい

Question

実現したいこと

sqlite3に入れたデータをデータフレームに変換し、pandasを使いデータを整理しています。

df
index |カラム1|カラム2|カラム3|
0 |A|B|C|
1 |B|A|C|※A,B,Cは文字列
というデータフレームがあり、カラム1とカラム2にAが含まれている行を抽出したいと思ったときに、

df[df["カラム1"] == "A"
|[df["カラム2"] == "A"]

とすれば、抽出はできるのですが、カラムが複数になるともっと短く実現できないかな…と考えおります。

発生している問題・分からないこと

sql文では
"select count(*) from table" where ' 文字列' IN (カラム1,カラム2,カラム3)"

というwhere in を使って複数のカラムから同じ文字列があるデータを抽出することができました。

df.query("カラム1 in ['A','B']

みたいな感じで

df.query("'A' in [カラム1,カラム2]

としてみたところ、SyntaxErrorが出てしまいます。

該当のソースコード

特になし

試したこと・調べたこと

• teratailやGoogle等で検索した
• ソースコードを自分なりに変更した
• 知人に聞いた
• その他

上記の詳細・結果

sqlとpandasの対応表の記事を読み、where in に相当する書き方を試してみたのですが、エラーが出てしまいます。

補足

特になし

Answer 1

カラム1とカラム2にAが含まれている列を抽出したい

「含まれている列」と書かれていますが、これは「含まれている行」ではないでしょうか。その場合は any() メソッドを利用するとよいかと思います。

import pandas as pd

df = pd.DataFrame({
    'カラム1': ['A', 'B', 'D'],
    'カラム2': ['B', 'A', 'E'],
    'カラム3': ['C', 'C', 'A'],
})

print(df)
cols = ["カラム1", "カラム2"]
idx = df[cols].eq("A").any(axis=1)
print(df[idx])

#   カラム1 カラム2 カラム3
# 0       A       B       C
# 1       B       A       C
# 2       D       E       A
#
#   カラム1 カラム2 カラム3
# 0       A       B       C
# 1       B       A       C

Comments

[naruyan_umpire]

すみません、ご指摘通り行です！！
試したところうまくできました、ご回答ありがとうございました！！
ベストアンサーに選ばせていただきました。