def METHOD_NAME(cls, ctx, op: "DataFrameExplode"):
    in_data = ctx[op.inputs[0].key]
    if in_data.ndim == 2:
        ctx[op.outputs[0].key] = in_data.explode(op.column)
    else:
        ctx[op.outputs[0].key] = in_data.explode()