pandasで指定の行のみ値を計算する方法

Question

お世話になります。質問させていただきます。

pandasをつかって、指定の行のみ計算したいです。

やりたいことは
こちらに書いてある「loc, ilocでブールインデックス参照」
もしくは
こちらに書いてあるstr.contains()
を組合せるのかなとおもっております。

やりたいことは

AAA	AAC	BBB
0	1	5
-5	-7	-2
0.5	7	0.1

というデータがあったとき、AA*のカラム名（AAAとAAC）の値のみ調べ、
値が0以下だった場合は－100とし、値が0以上だったら100とするような
ことをしたいです。

そこで、

df.loc[df.columns.str.contains('AA') < 0, df.columns.str.contains('AA') )] = -100

としてみましたが、何も変化はありませんでした。

どなたか教えていただけますとありがたいです。

どうぞよろしくお願い致します。

Answer 1

ループを使う方法

for col in df.columns[df.columns.str.match('AA.')]:
    df[col] = df[col].apply(lambda d:100 if d>0 else -100)
print(df)

dataframe.update()を使う方法

tmp = df.applymap(lambda d:100 if d>0 else -100)
df.loc[:, df.columns.str.match('AA.')] = np.nan
df.update(tmp, overwrite=False)
print(df)

Answer 2

自己解決できました。

１．pandasでcsvファイルを読み込む
２．カラム名のみ抽出
３．カラム名の中でほしい文字列のもののみ抽出してリストにする
４．リストにしたものをloc, ilocでブールインデックス参照を使って処理する。

上記のやり方で解決しました。

下記コードです。

import pandas as pd

df = pd.read_csv('hogehoge.csv',encoding='shift-jis',engine='python')

cols_list = list(df.columns)
focus_list = [s for s in cols_list if 'hogehoge(※抽出したカラム名に含まれている文字列)' in s]

for i in focus_list:
    df.loc[df[i] > 0, i] = 100
    df.loc[~(df[i] > 0) , i]=-100

df.to_csv('result.csv')

もっといい方法があれば教えていただけると嬉しいです。
どうぞよろしくお願いします。