[Python]pandas データフレームの列要素と変数の比較、抜き出し

Question

前提・実現したいこと

下記のようなcsvがあるとします

dataset	number	tag
A	0	1m
	1	2m
B	0	1m
	2	3m
	5	4m

dataset列の値と変数の中身が等しい時に
該当箇所を抜き出したいです。
お願い致します。

ex) temp_data = "A"
dataset列に"A"は含まれるので

dataset	number	tag
A	0	1m
	1	2m

該当のソースコード

>>> import pandas as pd
>>> import io
>>> csv = '''
... dataset,number,tag
... A,0,1m
... ,1,2m
... B,0,1m
... ,2,3m
... ,5,4m
... '''
>>> df = pd.read_csv(io.StringIO(csv))
>>> df
  dataset  number tag
0       A       0  1m
1     NaN       1  2m
2       B       0  1m
3     NaN       2  3m
4     NaN       5  4m

# 欠損値を埋める
>>> df2 = df.fillna(method="ffill")
>>> df2
  dataset  number tag
0       A       0  1m
1       A       1  2m
2       B       0  1m
3       B       2  3m
4       B       5  4m
temp_data = "A" 
#この時以下のように抜き出したい..
  dataset  number tag
0       A       0  1m
1       A       1  2m

動作環境

Windows10
Python3.6.5
pandas 0.24.2

Answer 1

df2.loc[df2['dataset']=='A'] で良いかと思います。

import pandas as pd
import io
csv = '''
dataset,number,tag
A,0,1m
,1,2m
B,0,1m
,2,3m
,5,4m
'''
df = pd.read_csv(io.StringIO(csv)).fillna(method="ffill")

print(df.loc[df['dataset']=='A'])
#  dataset  number tag
#0       A       0  1m
#1       A       1  2m

Comments

[zeze]

お世話になっております。解決いたしました。ありがとうございました！！