prerna.reactor.frame.py.TransposeReactor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation
SEMOSS
The newest version!
package prerna.reactor.frame.py;

import prerna.ds.py.PandasFrame;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.PixelOperationType;
import prerna.sablecc2.om.nounmeta.NounMetadata;
import prerna.util.usertracking.AnalyticsTrackerHelper;
import prerna.util.usertracking.UserTrackerFactory;

public class TransposeReactor extends AbstractPyFrameReactor {

	@Override
	public NounMetadata execute() {
		PandasFrame frame = (PandasFrame) getFrame();
		// get table name
		String table = frame.getWrapperName();

		String transposeScript = frame.getName() + "=" + table + ".cache['data'].transpose().rename_axis('rn', axis=0).add_prefix('V').reset_index()";
		// python transpose creates numeric columns
		// need to make str type
		// change all datatypes to string
		String objectType = frame.getName() + "=" + frame.getName() + ".astype(str)";
		String stringType = frame.getName() + "=" + frame.getName() + ".astype('string')";
		insight.getPyTranslator().runEmptyPy(transposeScript, objectType, stringType);
		this.addExecutedCode(transposeScript);
		this.addExecutedCode(objectType);
		this.addExecutedCode(stringType);

		//String[] colTypes = getColumnTypes(frame);
		//insight.getPyTranslator().runPyAndReturnOutput(transposeScript, stringHeaderType, stringType);

		// TRACKING
		UserTrackerFactory.getInstance().trackAnalyticsWidget(this.insight, frame, "Transpose", AnalyticsTrackerHelper.getHashInputs(this.store, this.keysToGet));
		// the column data has changed
		frame = (PandasFrame) recreateMetadata(frame);
		return new NounMetadata(frame, PixelDataType.FRAME, PixelOperationType.FRAME_DATA_CHANGE, PixelOperationType.FRAME_HEADERS_CHANGE);
	}
	
//	@Override
//	public NounMetadata execute() {
//		PandasFrame frame = (PandasFrame) getFrame();
//		// get table name
//		String table = frame.getName();
//		String transposeScript = table + " = " + table + ".transpose()";
//		// python transpose creates numeric columns
//		// need to make str type
//		String stringHeaderType = table + ".columns = " + table + ".columns.astype(str)";
//		// change all datatypes to string
//		String stringType = table + " = " + table + ".astype(str)";
//		insight.getPyTranslator().runPyAndReturnOutput(transposeScript, stringHeaderType, stringType);
//		// TRACKING
//		UserTrackerFactory.getInstance().trackAnalyticsWidget(this.insight, frame, "Transpose", AnalyticsTrackerHelper.getHashInputs(this.store, this.keysToGet));
//		// the column data has changed
//		frame = (PandasFrame) recreateMetadata(frame);
//		return new NounMetadata(frame, PixelDataType.FRAME, PixelOperationType.FRAME_DATA_CHANGE, PixelOperationType.FRAME_HEADERS_CHANGE);
//	}

}