pandas DataFrameで条件に一致する列名の取得

Question

インデックス、複数の列名、複数の値が一致する、の条件を満たす列名を取得したいです。

以下のDFで、

   index                      time    open    high     low   close  volume     ThreeBlackCrows  ThreeInside  ThreeLineStrike  ThreeOutside  
0      0 2010-01-04 00:00:00+00:00  92.748  92.964  92.748  92.898  1701.0                  0            0                0             0   
1      1 2010-01-04 01:00:00+00:00  92.900  93.076  92.894  92.980  1263.0                  0            0                0             0   
2      2 2010-01-04 02:00:00+00:00  92.982  92.984  92.845  92.874   879.0                  0            0                0             0   
3      3 2010-01-04 03:00:00+00:00  92.872  92.918  92.724  92.757   762.0                  0            0                0          -100   
4      4 2010-01-04 04:00:00+00:00  92.760  92.793  92.726  92.745   463.0                  0            0             -200          -100   

このDFに対して、
インデックス：３
列名：ThreeBlackCrows ThreeInside ThreeLineStrike ThreeOutside
値：-100、-200
を指定しThreeOutsideを取得

インデックス：４
列名：ThreeBlackCrows ThreeInside ThreeLineStrike ThreeOutside
値：-100、-200
を指定し、ThreeLineStrike ThreeOutsideを取得

という処理を作りたいです。

追記：
値の条件が100,200のパターンもあり、open等の数値が100にならないとは限らないです。

試したのは

パターン１
df=df.loc[idx,["TwoCrows", "ThreeBlackCrows", "ThreeInside", "ThreeLineStrike", "ThreeOutside"]].isin([-100,-200])

結果：
TwoCrows           False
ThreeBlackCrows    False
ThreeInside        False
ThreeLineStrike    False
ThreeOutside        True
Name: 150, dtype: bool

パターン２
df = df[idx:idx+1][["TwoCrows", "ThreeBlackCrows", "ThreeInside", "ThreeLineStrike", "ThreeOutside"]].isin([-100,-200])

結果：
     TwoCrows  ThreeBlackCrows  ThreeInside  ThreeLineStrike  ThreeOutside
150     False            False        False            False          True


パターン３
df = df[idx:idx+1][["TwoCrows", "ThreeBlackCrows", "ThreeInside", "ThreeLineStrike", "ThreeOutside"]].isin([-100,-200]).columns.values

結果：
['TwoCrows' 'ThreeBlackCrows' 'ThreeInside' 'ThreeLineStrike'
 'ThreeOutside']

条件に一致する列名を取得する方法お分かりになれば教えて頂きたいです。
よろしくお願いいたします。

Answer 1

すでにdfにデータが設定されている前提で、

cols = ['ThreeBlackCrows', 'ThreeInside', 'ThreeLineStrike', 'ThreeOutside']

idx = 3
s = df.loc[idx, cols].isin([-100, -200])
answer = ",".join(s[s == True].index.values)
print(f"index: {idx}: {answer}")

idx = 4
s = df.loc[idx, cols].isin([-100, -200])
answer = ",".join(s[s == True].index.values)
print(f"index: {idx}: {answer}")

index: 3: ThreeOutside
index: 4: ThreeLineStrike,ThreeOutside

Comments

[666_paru]

できました！
助かりました。

ありがとうございます。