Python 使用多个标签拆分数据帧

Python 使用多个标签拆分数据帧,python,pandas,dataframe,numpy,subset,Python,Pandas,Dataframe,Numpy,Subset,我正在尝试按不同的标签分割数据,如下所示: dfa=dfu a[(df_a['label']==0)|(df_a['label']==15)|(df_a['label']==16))] 这对于少量的数字来说效果很好。但是,我想对许多值执行此操作。 例如: to_train=(1,2,3,4,5,6,7,8,9,10,11,12,13,14,17,18,19,20)#这是可以改变的 如果df_b['label']==i]#值错误,则dfb=[i代表i进入列车 这就产生了一个错误: ValueE

我正在尝试按不同的标签分割数据,如下所示:

dfa=dfu a[(df_a['label']==0)|(df_a['label']==15)|(df_a['label']==16))]
这对于少量的数字来说效果很好。但是,我想对许多值执行此操作。 例如:

to_train=(1,2,3,4,5,6,7,8,9,10,11,12,13,14,17,18,19,20)#这是可以改变的
如果df_b['label']==i]#值错误,则dfb=[i代表i进入列车
这就产生了一个错误:

ValueError:序列的真值不明确。使用a.empty、a.bool()、a.item()、a.any()或a.all()

我已经读过这个错误,但我已经使用了位运算符,根据我的理解,它们不能解决很多问题


如何根据元组/列表/等中的内容拆分数据帧?

这是可行的,只需更改为
dfb=df_b[df_b['label'].isin(to_train)]
,专门针对我的代码。
to_train = (1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,17, 18, 19, 20)
dfb = dfa[df_a['label'].isin(to_train)]