com.github.chen0040.data.frame.OutputDataColumn Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of java-data-frame Show documentation

Some common patterns of data frame in Java

The newest version!

package com.github.chen0040.data.frame;


import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;


/**
 * Created by xschen on 5/5/2017.
 */
public class OutputDataColumn implements Serializable, DataColumn {
   private static final long serialVersionUID = -3482432625139406828L;
   private String columnName;
   private final List levels = new ArrayList<>();

   public OutputDataColumn(){

   }

   public OutputDataColumn(String columnName) {
      this.columnName = columnName;
   }

   @Override
   public String getColumnName(){
      return columnName;
   }

   @Override
   public void setColumnName(String columnName) {
      this.columnName = columnName;
   }

   @Override
   public boolean isCategorical(){
      return !levels.isEmpty();
   }


   public OutputDataColumn makeCopy() {
      OutputDataColumn clone = new OutputDataColumn(columnName);
      clone.copy(this);
      return clone;
   }

   @Override
   public void setLevels(List levels) {
      this.levels.clear();
      this.levels.addAll(levels);
   }

   @Override
   public List getLevels(){
      return levels;
   }

   public void copy(OutputDataColumn that){
      columnName = that.columnName;
      levels.clear();
      levels.addAll(that.levels);
   }

   @Override
   public boolean isOutputColumn(){
      return true;
   }
}