导入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