我有以下pandas DataFrame名为df_trainers
:
name trainer
0 BINARY MLPClassifier(activation='tanh', alpha=0.0001,...
1 REG MLPRegressor(activation='tanh', alpha=0.0001, ...
2 SVM SVC(C=1.0, cache_size=200, class_weight=None, ...
我希望获得与特定名称相对应的培训师。 例如
name='SVM'
tr = df_trainers[df_trainers['name'] == name]['trainer']
但它说tr的类型是<class 'pandas.core.series.Series'>
而不是<class 'sklearn.svm.classes.SVC'>
如何访问培训师本身而不是将其作为一个系列?
答案 0 :(得分:3)
对于Series
的标量选择值,您似乎得到一个项目[]
(如果boolean indexing
返回更多行,则可能会更改values[1]...
以选择第二项{ {1}}):
Series
或者:
tr = df_trainers.loc[df_trainers['name'] == name, 'trainer'].values[0]
或者:
tr = df_trainers.loc[df_trainers['name'] == name, 'trainer'].item()