Python 根据给定的顺序值/编号查询并截取数据表中指定行的数据

若X为特征数据，y为观测值数据【单列】，若要实现根据指定的值/编号查询并截取数据表中指定行的数据，代码实现如下：

def data_subject(X,y,sub_map,sub_id):
   ix=[i for i in range(len(sub_map)) if sub_map[i]==sub_id]
   return X[ix,;,;],y[ix]

其中sub为目标列所有行数据集合；sub_id为指定的编号/值。