.locのセル情報を抽出する方法

Question

前提・実現したいこと

Pythonで統計データを算出するプログラムを作成中です。
pandasのDataFrameを用いて、任意の列に"b"という文字列が入っている場合に
"bを検知"という文字列を出力したいと考えております。

今後、文字列"b"を検知した際に"A"列の値を用いて統計量を算出したいと考えています。
そのため、ブールインデックス参照などを用いた解決ではなく、
if文を動作させられる助言がいただけますと幸いです。

発生している問題・エラーメッセージ

該当のソースコードでは"bを検知"という文字列が出力されない。

該当のソースコード

df_test = pd.DataFrame({'A': [-20, -10, 0, 10, 20],
                   'B': [1, 2, 3, 4, 5],
                   'C': ['a', 'b', 'b', 'b', 'a']})

for itr in range(0, len(df_test), 1):
	print(itr)
	if str(df_test.loc[itr:itr, "C"]) == "b":
		print("bを検知")

試したこと

print(df_test.loc[1:1, "C"])を出力し、
"b"という文字列を出力できることが分かりました。

ただ、dtype: objectと出力結果に出ていたので、
if文ではstr()で型を文字列型に変換しましたがうまく出力されませんでした。

補足情報（FW/ツールのバージョンなど）

Python3.6
Visual Studio 2019

Answer 1

df_test.loc[itr:itr, "C"]で得られるのはシリーズ、つまり長さ1の配列です。
ここでほしいのは配列の要素ですので、該当部分を

df_test.loc[itr, "C"]

とするか、さらに最適な書き方としては

df_test.at[itr, "C"]

のように修正します。

Comments

[chocotruffe]

回答ありがとうございました。
解決できました。

Answer 2

df_test.loc[itr:itr, "C"] は pandas.Series 型のインスタンスになりますので、スカラー値(この倍は文字列)を取り出す場合は以下の様にします。

    #if str(df_test.loc[itr:itr, "C"]) == "b":
    if df_test.loc[itr, "C"] == "b":

Comments

[chocotruffe]

回答ありがとうございました。
解決できました。