mixly3/boards/default_src/python_mixpy/generators/data.js

import * as Blockly from 'blockly/core';

export const series_create = function (_, generator) {
    generator.definitions_.import_pandas = "import pandas";
    var varName1 = generator.valueToCode(this, 'SER', generator.ORDER_ASSIGNMENT) || '0';
    var varName = generator.variableDB_.getName(this.getFieldValue('VAR'),
        Blockly.Variables.NAME_TYPE);
    var code = varName + ' = ' + 'pandas.Series(' + varName1 + ')\n';
    return code;
}

export const series_create_from_index = function (_, generator) {
    generator.definitions_.import_pandas = "import pandas";
    var varName1 = generator.valueToCode(this, 'SER', generator.ORDER_ASSIGNMENT) || '0';
    var varName2 = generator.valueToCode(this, 'INDEX', generator.ORDER_ATOMIC) || '\'\'';
    var varName = generator.variableDB_.getName(this.getFieldValue('VAR'),
        Blockly.Variables.NAME_TYPE);
    var code = varName + ' = ' + 'pandas.Series(' + varName1 + ',' + 'index=' + varName2 + ')\n';
    return code;
}

export const dataframe_create = function (_, generator) {
    generator.definitions_.import_pandas = "import pandas";
    var varName1 = generator.valueToCode(this, 'SER', generator.ORDER_ASSIGNMENT) || '0';
    var varName = generator.variableDB_.getName(this.getFieldValue('VAR'),
        Blockly.Variables.NAME_TYPE);
    var code = varName + ' = ' + 'pandas.DataFrame(' + varName1 + ')\n';
    return code;
}

export const dataframe_create_from_index = function (_, generator) {
    generator.definitions_.import_pandas = "import pandas";
    var varName1 = generator.valueToCode(this, 'SER', generator.ORDER_ASSIGNMENT) || '0';
    var varName2 = generator.valueToCode(this, 'INDEX_COLUMN', generator.ORDER_ATOMIC) || '\'\'';
    var varName3 = generator.valueToCode(this, 'INDEX_RAW', generator.ORDER_ATOMIC) || '\'\'';
    var varName = generator.variableDB_.getName(this.getFieldValue('VAR'),
        Blockly.Variables.NAME_TYPE);
    var code = varName + ' = ' + 'pandas.DataFrame(' + varName1 + ',' + 'columns=' + varName2 + ',index=' + varName3 + ')\n';
    return code;
}

export const dataframe_create_from_one_index = function (_, generator) {
    generator.definitions_.import_pandas = "import pandas";
    var name = this.getFieldValue('COLUMN_RAW');
    var varName1 = generator.valueToCode(this, 'SER', generator.ORDER_ASSIGNMENT) || '0';
    var varName2 = generator.valueToCode(this, 'INDEX', generator.ORDER_ATOMIC) || '\'\'';
    var varName = generator.variableDB_.getName(this.getFieldValue('VAR'),
        Blockly.Variables.NAME_TYPE);
    var code = varName + ' = ' + 'pandas.DataFrame(' + varName1 + ',' + name + '=' + varName2 + ')\n';
    return code;
}

export const series_create_from_text = function (_, generator) {
    generator.definitions_.import_pandas = "import pandas";
    var varName = generator.variableDB_.getName(this.getFieldValue('VAR'),
        Blockly.Variables.NAME_TYPE);

    var text = this.getFieldValue('TEXT');
    var code = varName + ' = ' + 'pandas.Series([' + text + '])\n';
    return code;
}

export const series_index_value = function (_, generator) {
    generator.definitions_.import_pandas = "import pandas";
    var varName = generator.valueToCode(this, 'SERIES', generator.ORDER_ASSIGNMENT) || '0';
    var name = this.getFieldValue('INDEX_VALUE');
    var code = varName + '.' + name;
    return [code, generator.ORDER_ATOMIC];
}

export const series_get_num = function (_, generator) {
    // Indexing into a list is the same as indexing into a string.
    var varName = generator.valueToCode(this, 'SER', generator.ORDER_ASSIGNMENT) || '0';
    var argument0 = generator.valueToCode(this, 'AT',
        generator.ORDER_ADDITIVE) || '1';

    var code = varName + '[' + argument0 + ']';
    return [code, generator.ORDER_ATOMIC];
}

export const pl_show = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var code = 'pylab.show()\n';
    return code;
}

export const pl_axes = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var code = 'pylab.axes(aspect=1)\n';
    return code;
}

export const pl_plot_easy = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var varName = generator.valueToCode(this, 'SER', generator.ORDER_ASSIGNMENT) || '0';
    var code = 'pylab.plot(' + varName + ")\n";
    return code;
}

export const pl_plot = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var line = this.getFieldValue('LINE');
    var color = this.getFieldValue('COLOR');
    var dot = this.getFieldValue('DOT');
    var varName = generator.valueToCode(this, 'SER', generator.ORDER_ASSIGNMENT) || '0';
    var code = 'pylab.plot(' + varName + ",'" + dot + line + color + "')\n";
    return code;
}

export const pl_legend = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    generator.definitions_.import_matplotlib_font_manager = "import matplotlib.font_manager";
    var code = 'pylab.legend(' + 'prop=matplotlib.font_manager.FontProperties("' + "STSong" + '")' + ')\n';
    return code;
}

export const pl_title = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var a = generator.valueToCode(this, 'TITLE', generator.ORDER_ATOMIC);
    var code = 'pylab.title(' + a + ', fontproperties = "' + "STSong" + '")\n';
    return code;
}

export const pl_label = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var direction = this.getFieldValue('DIR');
    var a = generator.valueToCode(this, 'LABEL', generator.ORDER_ATOMIC);
    var code = 'pylab.' + direction + 'label(' + a + ', fontproperties = "' + "STSong" + '")\n';
    return code;
}

export const array_create = function (_, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var from = generator.valueToCode(this, "FROM", generator.ORDER_NONE) || "0";
    var end = generator.valueToCode(this, "TO", generator.ORDER_NONE) || "0";
    var step = generator.valueToCode(this, "STEP", generator.ORDER_NONE) || "1";
    var code = "numpy.arange(" + from + ", " + end + ", " + step + ")";
    return [code, generator.ORDER_ATOMIC];
}

export const pl_plot_bar = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var direction = this.getFieldValue('DIR');
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var code = 'pylab.' + direction + '(' + a + ',' + b + ")\n";
    return code;
}

export const pl_plot_scatter = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var s = generator.valueToCode(this, 'S', generator.ORDER_ATOMIC) || '\'\'';
    var color = this.getFieldValue('COLOR');
    var dot = this.getFieldValue('DOT');
    var tag = generator.valueToCode(this, 'TAG', generator.ORDER_ATOMIC) || '\'\'';
    var code = 'pylab.scatter(' + a + ',' + b + ",s=" + s + ",c='" + color + "',marker='" + dot + "',label=" + tag + ")\n";
    return code;
}

export const pl_plot_xy = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var line = this.getFieldValue('LINE');
    var color = this.getFieldValue('COLOR');
    var dot = this.getFieldValue('DOT');
    var tag = generator.valueToCode(this, 'TAG', generator.ORDER_ATOMIC) || '\'\'';
    var code = 'pylab.plot(' + a + ',' + b + ",'" + dot + line + color + "'" + ',label=' + tag + ")\n";
    return code;
}

export const pl_bar = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var tag = generator.valueToCode(this, 'TAG', generator.ORDER_ATOMIC) || '\'\'';
    var width = generator.valueToCode(this, 'WIDTH', generator.ORDER_RELATIONAL) || '0';
    var color = this.getFieldValue('COLOR')
    var align = this.getFieldValue('ALIGN');
    var code = 'pylab.bar(' + a + ',' + b + ',align="' + align + '",color="' + color + '",width=' + width + ',label=' + tag + ")\n";
    return code;
}

export const pl_pie = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var explode = generator.valueToCode(this, 'EXPLODE', generator.ORDER_ATOMIC) || '\'\'';
    var shadow = this.getFieldValue('SHADOW');
    var autopct = this.getFieldValue('autopct');
    if (autopct != 'None') { autopct = "'" + autopct + "'" }
    var code = 'pylab.pie(' + a + ',explode=' + explode + ',labels=' + b + ',autopct=' + autopct + ',shadow=' + shadow + ")\n";
    return code;
}

export const pl_hist = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var code = 'pylab.hist(' + a + ',' + b + ")\n";
    return code;
}

export const pl_ticks = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var direction = this.getFieldValue('DIR');
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var code = 'pylab.' + direction + 'ticks(' + a + ',' + b + ",fontproperties = '" + "STSong" + "')\n";
    return code;
}

export const numpy_trig = function (_, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var argument0 = generator.valueToCode(this, 'NUM', generator.ORDER_NONE) || '0';
    var operator = this.getFieldValue('OP');
    var code = "";

    code = "numpy." + operator + '(' + argument0 + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const pl_subplot = function (_, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var from = generator.valueToCode(this, "VET", generator.ORDER_NONE) || "0";
    var end = generator.valueToCode(this, "HOR", generator.ORDER_NONE) || "0";
    var step = generator.valueToCode(this, "NUM", generator.ORDER_NONE) || "0";
    var code = "pylab.subplot(" + from + ", " + end + ", " + step + ")\n";
    return code
}

export const pandas_readcsv = function (_, generator) {
    // For each loop.
    generator.definitions_.import_pandas = "import pandas";
    var fn = generator.valueToCode(this, 'FILENAME', generator.ORDER_ATOMIC);
    var mode = this.getFieldValue('MODE');
    var code = 'pandas.read_csv(' + fn + ', header=' + mode + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const dataframe_get = function (_, generator) {
    var mode = this.getFieldValue('MODE');
    var varName = generator.valueToCode(this, 'DICT', generator.ORDER_ASSIGNMENT) || '0';
    var text = generator.valueToCode(this, 'KEY', generator.ORDER_ASSIGNMENT);
    if (mode == "column") {
        var code = varName + "[" + text + "]";
    }
    else if (mode == 'raw') {
        var code = varName + ".loc[" + text + "]";
    }
    return [code, generator.ORDER_ATOMIC];
}

export const pl_savefig = function (_, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var file = generator.valueToCode(this, 'FILE', generator.ORDER_ATOMIC);
    var code = "pylab.savefig(" + file + ")\n";
    return code;
}

export const pl_text = function (_, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var from = generator.valueToCode(this, "VET", generator.ORDER_NONE) || "0";
    var end = generator.valueToCode(this, "HOR", generator.ORDER_NONE) || "0";
    var step = generator.valueToCode(this, "NUM", generator.ORDER_NONE) || "0";
    var halign = this.getFieldValue('HALIGN');
    var valign = this.getFieldValue('VALIGN');
    var fontnum = generator.valueToCode(this, 'FONTNUM', generator.ORDER_ASSIGNMENT) || '0';
    var code = "pylab.text(" + from + ", " + end + ", " + step + ", ha='" + halign + "', va='" + valign + "', fontsize=" + fontnum + ")\n";
    return code
}

export const array_toarray = function (_, generator) {
    var str = generator.valueToCode(this, 'VAR', generator.ORDER_ATOMIC) || '0';
    generator.definitions_['import_numpy'] = 'import numpy';
    var code = 'numpy.array(' + str + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const plot_show = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var code = 'matplotlib.pyplot.show()\n';
    return code;
}

export const plot_axes = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var code = 'matplotlib.pyplot.axes(aspect=1)\n';
    return code;
}

export const plot_plot_easy = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var varName = generator.valueToCode(this, 'SER', generator.ORDER_ASSIGNMENT) || '0';
    var code = 'matplotlib.pyplot.plot(' + varName + ")\n";
    return code;
}

export const plot_plot = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var line = this.getFieldValue('LINE');
    var color = this.getFieldValue('COLOR');
    var dot = this.getFieldValue('DOT');
    var varName = generator.valueToCode(this, 'SER', generator.ORDER_ASSIGNMENT) || '0';
    var code = 'matplotlib.pyplot.plot(' + varName + ",'" + dot + line + color + "')\n";
    return code;
}

export const plot_legend = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    generator.definitions_.import_matplotlib_font_manager = "import matplotlib.font_manager";
    var code = 'matplotlib.pyplot.legend(' + 'prop=matplotlib.font_manager.FontProperties("' + "STSong" + '")' + ')\n';
    return code;
}

export const plot_title = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var a = generator.valueToCode(this, 'TITLE', generator.ORDER_ATOMIC);
    var code = 'matplotlib.pyplot.title(' + a + ', fontproperties = "' + "STSong" + '")\n';
    return code;
}

export const plot_label = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var direction = this.getFieldValue('DIR');
    var a = generator.valueToCode(this, 'LABEL', generator.ORDER_ATOMIC);
    var code = 'matplotlib.pyplot.' + direction + 'label(' + a + ', fontproperties = "' + "STSong" + '")\n';
    return code;
}

export const plot_plot_bar = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var direction = this.getFieldValue('DIR');
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var code = 'matplotlib.pyplot.' + direction + '(' + a + ',' + b + ")\n";
    return code;
}

export const plot_plot_scatter = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var s = generator.valueToCode(this, 'S', generator.ORDER_ATOMIC) || '\'\'';
    var color = this.getFieldValue('COLOR');
    var dot = this.getFieldValue('DOT');
    var tag = generator.valueToCode(this, 'TAG', generator.ORDER_ATOMIC) || '\'\'';
    var code = 'matplotlib.pyplot.scatter(' + a + ',' + b + ",s=" + s + ",c='" + color + "',marker='" + dot + "',label=" + tag + ")\n";
    return code;
}

export const plot_plot_xy = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var line = this.getFieldValue('LINE');
    var color = this.getFieldValue('COLOR');
    var dot = this.getFieldValue('DOT');
    var tag = generator.valueToCode(this, 'TAG', generator.ORDER_ATOMIC) || '\'\'';
    var code = 'matplotlib.pyplot.plot(' + a + ',' + b + ",'" + dot + line + color + "'" + ',label=' + tag + ")\n";
    return code;
}

export const plot_bar = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var tag = generator.valueToCode(this, 'TAG', generator.ORDER_ATOMIC) || '\'\'';
    var width = generator.valueToCode(this, 'WIDTH', generator.ORDER_RELATIONAL) || '0';
    var color = this.getFieldValue('COLOR')
    var align = this.getFieldValue('ALIGN');
    var code = 'matplotlib.pyplot.bar(' + a + ',' + b + ',align="' + align + '",color="' + color + '",width=' + width + ',label=' + tag + ")\n";
    return code;
}

export const plot_pie = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var explode = generator.valueToCode(this, 'EXPLODE', generator.ORDER_ATOMIC) || '\'\'';
    var shadow = this.getFieldValue('SHADOW');
    var autopct = this.getFieldValue('autopct');
    if (autopct != 'None') { autopct = "'" + autopct + "'" }
    var code = 'matplotlib.pyplot.pie(' + a + ',explode=' + explode + ',labels=' + b + ',autopct=' + autopct + ',shadow=' + shadow + ")\n";
    return code;
}

export const plot_hist = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var code = 'matplotlib.pyplot.hist(' + a + ',' + b + ")\n";
    return code;
}

export const plot_ticks = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var direction = this.getFieldValue('DIR');
    var a = generator.valueToCode(this, 'A', generator.ORDER_ATOMIC) || '\'\'';
    var b = generator.valueToCode(this, 'B', generator.ORDER_ATOMIC) || '\'\'';
    var code = 'matplotlib.pyplot.' + direction + 'ticks(' + a + ',' + b + ",fontproperties = '" + "STSong" + "')\n";
    return code;
}

export const plot_subplot = function (_, generator) {
    generator.definitions_.import_numpy = "import numpy";
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var from = generator.valueToCode(this, "VET", generator.ORDER_NONE) || "0";
    var end = generator.valueToCode(this, "HOR", generator.ORDER_NONE) || "0";
    var step = generator.valueToCode(this, "NUM", generator.ORDER_NONE) || "0";
    var code = "matplotlib.pyplot.subplot(" + from + ", " + end + ", " + step + ")\n";
    return code
}

export const plot_savefig = function (_, generator) {
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var code = "matplotlib.pyplot.savefig('1.png')\n";
    return code;
}

export const plot_text = function (_, generator) {
    generator.definitions_.import_numpy = "import numpy";
    generator.definitions_.import_matplotlib_pyplot = "import matplotlib.pyplot";
    var from = generator.valueToCode(this, "VET", generator.ORDER_NONE) || "0";
    var end = generator.valueToCode(this, "HOR", generator.ORDER_NONE) || "0";
    var step = generator.valueToCode(this, "NUM", generator.ORDER_NONE) || "0";
    var halign = this.getFieldValue('HALIGN');
    var valign = this.getFieldValue('VALIGN');
    var fontnum = generator.valueToCode(this, 'FONTNUM', generator.ORDER_ASSIGNMENT) || '0';
    var code = "matplotlib.pyplot.text(" + from + ", " + end + ", " + step + ", ha='" + halign + "', va='" + valign + "', fontsize=" + fontnum + ")\n";
    return code
}

export const numpy_shape = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var code = 'numpy.shape(' + array + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_reshape = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var dimensions = generator.valueToCode(block, 'DIMENSIONS', generator.ORDER_ATOMIC) || '(1, -1)';
    var code = 'numpy.reshape(' + array + ', ' + dimensions + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_astype = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var dtype = block.getFieldValue('TYPE');
    var code = array + '.astype(' + dtype + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const pylab_imshow = function (block, generator) {
    generator.definitions_.import_pylab = "import pylab";
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var colormap = block.getFieldValue('COLORMAP') || 'gray';
    var code = 'pylab.imshow(' + array + ', cmap="' + colormap + '")\n';
    return code;
}

export const numpy_mean_digit = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var X = generator.valueToCode(block, 'X', generator.ORDER_ATOMIC) || '[]';
    var y = generator.valueToCode(block, 'Y', generator.ORDER_ATOMIC) || '[]';
    var digit = generator.valueToCode(block, 'DIGIT', generator.ORDER_ATOMIC) || '0';
    var code = 'numpy.mean(' + X + '[' + y + ' == ' + digit + '], axis=0)';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_argmin_template_match = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    generator.definitions_.import_cv2 = "import cv2";
    var image = generator.valueToCode(block, 'IMAGE', generator.ORDER_ATOMIC) || '[]';
    var templates = generator.valueToCode(block, 'TEMPLATES', generator.ORDER_ATOMIC) || '[]';
    var code = 'numpy.argmin([numpy.sum(cv2.absdiff(' + image + ', template)) for template in ' + templates + '])';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_mean = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var axis = block.getFieldValue('AXIS');
    if (axis === 'NONE') {
        var code = 'numpy.mean(' + array + ')';
    } else {
        var code = 'numpy.mean(' + array + ', axis=' + axis + ')';
    }
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_std = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var axis = block.getFieldValue('AXIS');
    if (axis === 'NONE') {
        var code = 'numpy.std(' + array + ')';
    } else {
        var code = 'numpy.std(' + array + ', axis=' + axis + ')';
    }
    return [code, generator.ORDER_ATOMIC];
}

export const pandas_dropna = function (block, generator) {
    generator.definitions_.import_pandas = "import pandas";
    var df = generator.valueToCode(block, 'DF', generator.ORDER_ATOMIC) || 'df';

    var code = df + '.dropna()';

    return [code, generator.ORDER_MEMBER];
}

export const pandas_drop_columns = function (block, generator) {
    generator.definitions_.import_pandas = "import pandas";
    var dataframe = generator.valueToCode(block, 'DATAFRAME', generator.ORDER_ATOMIC) || 'df';
    var columns = generator.valueToCode(block, 'COLUMNS', generator.ORDER_ATOMIC) || '[]';
    var axis = block.getFieldValue('AXIS') || '0';
    var code = dataframe + '.drop(columns=' + columns + ', axis=' + axis + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_ones = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var num = generator.valueToCode(block, 'NUM', generator.ORDER_ATOMIC) || '1';
    var code = 'numpy.ones(' + num + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_c_ = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var array1 = generator.valueToCode(block, 'ARRAY1', generator.ORDER_ATOMIC) || '[]';
    var array2 = generator.valueToCode(block, 'ARRAY2', generator.ORDER_ATOMIC) || '[]';
    var code = 'numpy.c_[' + array1 + ', ' + array2 + ']';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_linalg_det = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    generator.definitions_.import_linalg = "import numpy.linalg";
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var code = 'numpy.linalg.det(' + array + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const matrix_transpose = function (block, generator) {
    var matrix = generator.valueToCode(block, 'MATRIX', generator.ORDER_ATOMIC) || '[]';
    var code = matrix + '.T';
    return [code, generator.ORDER_ATOMIC];
}

export const matrix_multiplication = function (block, generator) {
    var matrix1 = generator.valueToCode(block, 'MATRIX1', generator.ORDER_ATOMIC) || '[]';
    var matrix2 = generator.valueToCode(block, 'MATRIX2', generator.ORDER_ATOMIC) || '[]';
    var code = matrix1 + ' @ ' + matrix2;
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_linalg_inv = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    generator.definitions_.import_linalg = "import numpy.linalg";
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var code = 'numpy.linalg.inv(' + array + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const dataframe_median = function (block, generator) {
    var dataframe = generator.valueToCode(block, 'DATAFRAME', generator.ORDER_ATOMIC) || 'df';
    var code = dataframe + '.median()';
    return [code, generator.ORDER_ATOMIC];
}

export const dataframe_fillna = function (block, generator) {
    var dataframe = generator.valueToCode(block, 'DATAFRAME', generator.ORDER_ATOMIC) || 'df';
    var value = generator.valueToCode(block, 'VALUE', generator.ORDER_ATOMIC) || '0';
    var inplace = block.getFieldValue('INPLACE') || 'False';
    var code = dataframe + '.fillna(value=' + value + ', inplace=' + inplace + ')\n';
    return code;
}

export const dataframe_info = function (block, generator) {
    var dataframe = generator.valueToCode(block, 'DATAFRAME', generator.ORDER_ATOMIC) || 'df';
    var code = dataframe + '.info()';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_min = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var axis = block.getFieldValue('AXIS');
    var code = (axis === 'None') ? 'numpy.min(' + array + ')' : 'numpy.min(' + array + ', axis=' + axis + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_max = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var axis = block.getFieldValue('AXIS');
    var code = (axis === 'None') ? 'numpy.max(' + array + ')' : 'numpy.max(' + array + ', axis=' + axis + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_size = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var code = array + '.size';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_dot = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var array1 = generator.valueToCode(block, 'ARRAY1', generator.ORDER_ATOMIC) || '[]';
    var array2 = generator.valueToCode(block, 'ARRAY2', generator.ORDER_ATOMIC) || '[]';
    var code = 'numpy.dot(' + array1 + ', ' + array2 + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_square = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var code = 'numpy.square(' + array + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_sum = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var code = 'numpy.sum(' + array + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const numpy_random_random = function (block, generator) {
    generator.definitions_.import_numpy = "import numpy";
    var size = generator.valueToCode(block, 'SIZE', generator.ORDER_ATOMIC) || '1';
    var code = 'numpy.random.random(' + size + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const py_sum = function (block, generator) {
    var array = generator.valueToCode(block, 'ARRAY', generator.ORDER_ATOMIC) || '[]';
    var code = 'sum(' + array + ')';
    return [code, generator.ORDER_ATOMIC];
}

export const dataframe_sort_values = function (block, generator) {
    const dict = generator.valueToCode(block, 'DICT', generator.ORDER_ATOMIC) || 'df';
    const key = generator.valueToCode(block, 'KEY', generator.ORDER_ATOMIC) || '\'tag\'';
    const ascending = block.getFieldValue('AS_CENDING');
    const code = `${dict}.sort_values(by=${key}, ascending=${ascending})`;
    return [code, generator.ORDER_ATOMIC];
}

export const dataframe_head_tail = function (block, generator) {
    const dict = generator.valueToCode(block, 'DICT', generator.ORDER_ATOMIC) || 'df';
    const lines = generator.valueToCode(block, 'LINES', generator.ORDER_ATOMIC) || '1';
    const type = block.getFieldValue('TYPE');
    const code = `${dict}.${type}(${lines})`;
    return [code, generator.ORDER_ATOMIC];
}

export const dataframe_select = function (block, generator) {
    const dict = generator.valueToCode(block, 'DICT', generator.ORDER_ATOMIC) || 'df';
    const key = generator.valueToCode(block, 'KEY', generator.ORDER_ATOMIC) || 'df[\'tag\'] > 1';
    const code = `${dict}[${key}]`;
    return [code, generator.ORDER_ATOMIC];
}

export const dataframe_groupby = function (block, generator) {
    const dict = generator.valueToCode(block, 'DICT', generator.ORDER_ATOMIC) || 'df';
    const key = generator.valueToCode(block, 'KEY', generator.ORDER_ATOMIC) || '\'tag\'';
    const code = `${dict}.groupby(by=${key})`;
    return [code, generator.ORDER_ATOMIC];
}

export const dataframe_aggregate_func = function (block, generator) {
    const dict = generator.valueToCode(block, 'DICT', generator.ORDER_ATOMIC) || 'df';
    const type = block.getFieldValue('TYPE');
    const code = `${dict}.${type}()`;
    return [code, generator.ORDER_ATOMIC];
}