リストにある文字列を含むDataFrameの行を取得したい

Question

前提・実現したいこと

以下のDataFrameがあるとします。

server id	os
001	Windows
002	CentOS
003	ubuntu
004	Windows
005	ubuntu

また、pythonのリスト「[001,003]」があるとします。
上記DataFrameのserver idのうち、このリストにある文字列を含むDataFrameの行のみを取得したいのですがどのようなやりかたがあるでしょうか。

期待する結果は以下です。

server id	os
001	Windows
003	ubuntu

よろしくお願いします。

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

CentOS 7
python 3.6

Answer 1

一応、Series.isin() というAPIもあります。

https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.Series.isin.html

import pandas as pd

df = pd.DataFrame({'server id': ['001','002','003','004','005'],
                   'os':['Windows','CentOS','ubuntu','Windows','ubuntu']})
id_list = ['001', '003']

ret = df.loc[df['server id'].isin(id_list)]
print(ret)
#  server id       os
#0       001  Windows
#2       003   ubuntu

server id が 重複がなく、Indexになっていているのでしたら、単純に Filterでもいけますね。

df = df.set_index('server id', drop=True)
ret = df.loc[id_list]
print(ret)
#                os
#server id
#001        Windows
#003         ubuntu

Comments

[user1]

ありがとうございました。大変参考になりました。

Answer 2

こんな感じじゃないですかね

# df: 元のデータフレーム
id_list = ['001', '003']
df[df['server id'] in id_list]

server idカラムとid_listは文字列としてます。

Comments

[user1]

ありがとうございました。
参考にさせて頂きます。