特定の文字列を含む行だけ抽出する際の詳細な指定

Question

Python3でPandasで特定の文字列を含む行のみを抽出しようとしています．
GG1とGG2のいずれかを含む行のみを抽出したいのですが，
GG10が抽出されてしまうので，
下のようにGG10を省く意図のプログラムを走らせたのですが，
下の出力結果のとおり，
GG2が抽出されなくなってしまいました．
当初，**id="GG1|GG2"**としていたら，GG2も抽出できていました．
解決策をご教示頂けますと幸いです．

id="GG1|GG2~GG10"
df[0].str.contains(id)
df[df[0].str.contains(id)].to_csv(out_file,header=False, index=False)

出力結果
GG1,1,2,57,14:50:05:JST,14:50:15:JST,10,11,2.199,2.199,,1,0,0,,
GG10,7,7,,,,,,,,,,,,,
GG1,1,3,65,14:52:41:JST,14:52:51:JST,10,11,2.199,2.199,,2,0,0,,
GG10,7,7,,,,,,,,,,,,,

Answer 1

import pandas as pd
import numpy as np

a = np.array(np.random.randint(1,10,50), dtype=np.str)
df = pd.DataFrame(a).apply(lambda x: 'GG'+x).rename(columns={0:'cols'})

df.query('cols in ["GG1","GG2"]')

pandas.DataFrame.query()でこのようにリスト抽出が可能です。

Comments

[LenItsuki]

できました．
rename(columns={0:'cols'})という感じで，特定のヘッダーをつける方法も分かりまして，
随分と処理をすすめることができます．
ありがとうございます．