matplotlib.pyplot.subplots.kt Maven / Gradle / Ivy

Go to download
package matplotlib.pyplot

import extensions.toPythonBooleanOrNone
import matplotlib.Axes
import matplotlib.Axes.Companion.axesNumber
import matplotlib.Figure
import matplotlib.Figure.Companion.figureNumber
import matplotlib.toKwargPythonDictOrNone
import python.PythonScriptBuilder

fun subplots(
    nrows: Int = 1,
    ncols: Int = 1,
    sharex: Boolean = false,
    sharey: Boolean = false,
    squeeze: Boolean = true,
    subplot_kw: Map? = null,
    gridspec_kw: Map? = null
): Pair {
    val figAxes = Pair(
        object : Figure {
            override val variableName: String = "figure_$figureNumber"
        },
        object : Axes {
            override val variableName: String = "axes_$axesNumber"
        }
    )
    PythonScriptBuilder.addCommand(
        "${figAxes.first.variableName}, ${figAxes.second.variableName} = plt.subplots(" +
                "nrows=$nrows," +
                "ncols=$ncols," +
                "sharex=${sharex.toPythonBooleanOrNone()}," +
                "sharey=${sharey.toPythonBooleanOrNone()}," +
                "squeeze=${squeeze.toPythonBooleanOrNone()}," +
                "subplot_kw=${subplot_kw.toKwargPythonDictOrNone()}," +
                "gridspec_kw=${gridspec_kw.toKwargPythonDictOrNone()}" +
                ")"
    )
    return figAxes
}