pandasの条件抽出がうまくいかない

Question

以下の文で2020-10-23の日付の7779の銘柄のtrading_unitを引き出したいと思っていますが、tがemptyとなってしまいます。
うまく拾う方法はありますでしょうか。

symbol = 7779
date = '2020-10-23'
t = df[(df["symbol"] == str(symbol)) & (df["created_date_at"] == str(date))]
if not t.empty:
    return t.iat[0, 1]
else:
    return 100

   symbol     trading_unit created_date_at
239 7779           100      2020-10-23

Comments

[meg_]

dfの各列の型は何ですか？

[taro_yamada]

以下の通りです。失礼しました。
symbol object
trading_unit int64
created_date_at object
dtype: object

[meg_]

object型となっているものがmixed typeとなっていて「7779」は数値として保持されている、という可能性は有りませんか？　念のためsymbol列をstr型に明示的に変換してみてはどうでしょうか？

[taro_yamada]

megさんのご指摘の通りでした。
created_date_atがstr型でなかったみたいで、seriesを明示的に文字列に変更したところうまく動きました。
ありがとうございました。

Answer 1

提示されたpandasのデータ結果において、列名とデータの縦位置がずれているのが気になります。
元データの7779（または2020-10-23）の前後に空白文字が含まれていないでしょうか？

その場合は、以下コードのようにすると正常に抽出できます。
またqueryを使うとシンプルに書けます。

import pandas as pd

df = pd.DataFrame({'symbol':['   7779   '], 'trading_unit':[100], 'created_date_at':['2020-10-23']})
df['symbol'] = df['symbol'].str.strip() # 前後に空白があれば取り除く

symbol = 7779
date = '2020-10-23'
t = df[(df["symbol"] == str(symbol)) & (df["created_date_at"] == str(date))]
print(t)
#   symbol  trading_unit created_date_at
#0    7779           100      2020-10-23

t = df.query(f'symbol == "{symbol}" and created_date_at == "{date}"')
print(t)
#   symbol  trading_unit created_date_at
#0    7779           100      2020-10-23

Comments

[taro_yamada]

すみません。しっかり書かなかったのですが、symbolはstrとして保持しています。
データ型は以下の通りです。
symbol object
trading_unit int64
created_date_at object
dtype: object

[8524ba23]

型について了解しました。回答を修正しました。

[taro_yamada]

やはりうまく動きません。
元の7779も同じDBから取り出してきているので、ずれることはなさそうです。

[8524ba23]

つまり前後に空白文字がついていることはないということでしょうか？

[taro_yamada]

はい。ついていることはないです。
.str.strip() も試しましたがemptyのままでした。

[8524ba23]

なるほど。あとは修正依頼欄でmeg_さんが指摘されている点を確認してみてどうか、でしょうか。

[taro_yamada]

megさんのご指摘の通りでした。
created_date_atがstr型でなかったみたいで、seriesを明示的に文字列に変更したところうまく動きました。
ありがとうございました。