python pandas dataframe locについて

以下のデータから　データを引数に渡してデータ取り出す方法を教えていただきたいです。

data1
__|年齢|性別|出身|
|:--|:--:|--:|
|19960101|24|男|東京
|19960101|36|男|神奈川
|19960102|45|女|埼玉
|19960102|40|男|大阪
|19960103|47|女|東京
|19960103|19|男|埼玉
|19960103|47|女|東京
|19960104|45|女|東京
|19960104|19|男|埼玉
|19960104|23|男|埼玉

上のdataframeの中から理想は

python
1
2a = [19960101, 19960103]
3data2 = data1.loc[a]

aのデータとdata1のindexが一致したデータを抽出したいです。
__|年齢|性別|出身|
|:--|:--:|--:|
|19960101|24|男|東京
|19960101|36|男|神奈川
|19960103|47|女|東京
|19960103|19|男|埼玉

data1のデータが大きいので上記には書けなかったのですが、
上記のpythonコードで一個一個は取り出せます。

for文やpd.concatでやろうとしたのですが、難しくてできませんでした。
教えていただけると嬉しいです。
よろしくお願いいたします。

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

回答1件

ベストアンサー

最初の行のcolumn名を'id'とした場合、以下でできます。

python
1a = [19960101, 19960103]
2print(df[df['id'].apply(lambda x: x in a)])

実行結果

python
1>>> a = [19960101, 19960103]
2>>> print(df)
3         id  年齢 性別   出身
40  19960101  24  男   東京
51  19960101  36  男  神奈川
62  19960102  45  女   埼玉
73  19960102  40  男   大阪
84  19960103  47  女   東京
95  19960103  19  男   埼玉
106  19960103  47  女   東京
117  19960104  45  女   東京
128  19960104  19  男   埼玉
139  19960104  23  男   埼玉
14>>> print(df[df['id'].apply(lambda x: x in a)])
15         id  年齢 性別   出身
160  19960101  24  男   東京
171  19960101  36  男  神奈川
184  19960103  47  女   東京
195  19960103  19  男   埼玉
206  19960103  47  女   東京