pandas通过列表选取DataFrame中特定的行

导入pandas

import pandas as pd

首先创建一个列表(列表中存储的是ID号）

a = [0,3,8,10]

然后创建一个DataFrame

data = {'ID':[0,1,2,3,4,5,6,7,8,9],
        'state': ['Ohio', 'Ohio', 'Ohio', 'Nevada', 'Nevada', 'Nevada', 'Ohio', 'Nevada', 'Ohio', 'Nevada'],
        'year': [2000, 2001, 2002, 2001, 2002, 2003, 2001, 2002, 2001, 2002],
        'pop': [1.5, 1.7, 3.6, 2.4, 2.9, 3.2, 1.5, 1.7, 3.6, 2.4]}

data = pd.DataFrame(data)
data





    ID	state	year	pop
0	0	Ohio	2000	1.5
1	1	Ohio	2001	1.7
2	2	Ohio	2002	3.6
3	3	Nevada	2001	2.4
4	4	Nevada	2002	2.9
5	5	Nevada	2003	3.2
6	6	Ohio	2001	1.5
7	7	Nevada	2002	1.7
8	8	Ohio	2001	3.6
9	9	Nevada	2002	2.4

通过a列表中的ID号在data中选取相对应的数据

data1 = data[(data['ID'].isin(a))]





    ID	state	year	pop
0	0	Ohio	2000	1.5
3	3	Nevada	2001	2.4
8	8	Ohio	2001	3.6