【データ分析】Pythonによるデータの抽出について

サーバーから取り出したチャットボットの発話のCSVを、ボット毎に分けた後、ユーザーとボットの発話に分けたいと考えています。

例えば以下のようなCSVがあった場合

・sendText: 発話内容
・userId: ユーザーのID
・botId: ボットのID
・messageType: ボットの発話ルール
・Bot: TrueがBotの発話 Falseがユーザーの発話

botIdの"B"のみを取り出し、その後Botの"True"のみを取り出したいとします。

python
1df_B=df[df.botId == "B"]

これで"B"のみを取り出すことができたのですが、その後に

python
1df_B_true=df_B[df_B.Bot == "True"]

とすると

/usr/local/lib/python3.6/dist-packages/pandas/core/ops/init.py:1115: FutureWarning: elementwise comparison failed; returning scalar instead, but in the future will perform elementwise comparison
result = method(y)

のメッセージが出て、headで確認すると１行目以外がすべてdropされてしまっています。
この場合はどういった処理が適切なのでしょうか？

行動規範の内容に同意します

回答1件

ベストアンサー

Bot列の型がstrではなくboolであると提示の警告が発生します。"True"ではなくTrueと比較すればよいです。

Python
1import pandas as pd
2from io import StringIO
3
4s = """sendText,userId,botId,messageTyep,Bot
5hello,abc,B,user,False
6hell?,abc,B,rule,True"""
7
8df = pd.read_csv(StringIO(s))
9print(df)
10#  sendText userId botId messageTyep    Bot
11#0    hello    abc     B        user  False
12#1    hell?    abc     B        rule   True
13
14print(df['Bot'].dtype) # bool
15
16df_B=df[df.botId == "B"]
17
18df_B_true=df_B[df_B.Bot == "True"] # FutureWarning: elementwise comparison failed; 
19
20df_B_true=df_B[df_B.Bot == True]
21print(df_B_true)
22#  sendText userId botId messageTyep   Bot
23#1    hell?    abc     B        rule  True