如何获取dataframe元素不像熊猫系列?

时间:2017-06-09 14:26:51

标签: pandas dataframe scikit-learn

我有以下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'>

如何访问培训师本身而不是将其作为一个系列?

1 个答案:

答案 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()