python dataframeから特定の文字列を含む行を判定し、抽出したい

前提・実現したいこと

以下の表のようなpandasのdataframeから、商品ラベル列で「100」が含まれている商品ラベル行に該当し
かつ、割引率の列が5である列のみを別のdataframeへ格納する方法を教えてください。

↓元のdataframeの中身

商品ラベル	値段(円)	割引率(%)
a_100_果物	300	5
a_101_食器	500	5
a_100_植物	100	5
a_103_果物	400	10
b_100_食器	500	15

上記のdataframeから

商品ラベル	値段(円)	割引率(%)
a_100_果物	300	5
a_100_植物	100	5

のような形でfataframeに抽出したいです

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

mac os
python 3.7

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

回答1件

ベストアンサー

Python
1import pandas as pd
2from io import StringIO
3
4s = """商品ラベル\t値段(円)\t割引率(%)
5a_100_果物\t300\t5
6a_101_食器\t500\t5
7a_100_植物\t100\t5
8a_103_果物\t400\t10
9b_100_食器\t500\t15"""
10
11df = pd.read_csv(StringIO(s), delimiter='\t')
12df['label'] = df['商品ラベル'].apply(lambda s: int(s.split('_')[1]))
13df = df[(df['割引率(%)'] == 5) & (df['label'] == 100)].drop(columns=['label']).reset_index(drop=True)
14print(df)
15#      商品ラベル  値段(円)  割引率(%)
16#0  a_100_果物    300       5
17#1  a_100_植物    100       5

投稿2020/07/30 14:10

編集2020/07/30 14:42

8524ba23

総合スコア38341

loreeeee

2020/07/30 14:34

回答いただきありがとうございます。プログラムを実行したところ df['label'] = df['商品ラベル'].apply(lambda s: int(s.split('_')[1])) File "/Users/pyenv/versions/anaconda3-5.3.1/envs/py37/lib/python3.7/site-packages/pandas/core/frame.py", line 2800, in __getitem__ indexer = self.columns.get_loc(key) File "/Users/pyenv/versions/anaconda3-5.3.1/envs/py37/lib/python3.7/site-packages/pandas/core/indexes/base.py", line 2648, in get_loc return self._engine.get_loc(self._maybe_cast_indexer(key)) File "pandas/_libs/index.pyx", line 111, in pandas._libs.index.IndexEngine.get_loc File "pandas/_libs/index.pyx", line 138, in pandas._libs.index.IndexEngine.get_loc File "pandas/_libs/hashtable_class_helper.pxi", line 1619, in pandas._libs.hashtable.PyObjectHashTable.get_item File "pandas/_libs/hashtable_class_helper.pxi", line 1627, in pandas._libs.hashtable.PyObjectHashTable.get_item KeyError: '商品ラベル' というkeyerrorと表示されました。なにが原因なのでしょうか？ pandasのバージョンは1.0.5です