在数据框列上应用select_dtypes

时间:2019-11-27 16:16:37

标签: python

我有2个数据框。其中一个是Train,另一个是Test。我列出了这两个数据框,以简化数据清理。现在我想对具有对象类型的列进行编码。问题在于,通过此代码,我可以选择对象类型的列,但不会将其应用于最终数据:

try {

    JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
    int timeout = dataSourceConfig.getTimeout();
    jdbcTemplate.setQueryTimeout(timeout);
    Result = jdbcTemplate.queryForObject(SQL, new Object[]{Params....}, String.class);

} catch(DataAccessException ex){
    log.error("Error Occurred :" + ex.getMessage());
} catch (Exception ex) {
    log.error("Error Occurred :" + ex.getMessage());

1 个答案:

答案 0 :(得分:1)

分配到相关列:

fulldata=[train,test]
for dataset in fulldata:
    object_cols = dataset.select_dtypes(include='object').columns
    dataset_transf = dataset[object_cols].apply(le.fit_transform)
    dataset[object_cols] = dataset_transf