def get_col_idxs(node_cols, ds_cols):
    """
    Get the index(es) of the input column(s) from the dataset
    """
    col_idxs = []
    non_exist_cols = []
    # temporary error if multiple node columns
    if len(node_cols) > 1:
        raise RuntimeError(
            "Offload hardware accelerator currently does not support map operations with multiple input columns")
    for node_col in node_cols:
        if node_col in ds_cols:
            col_idxs.append(ds_cols.index(node_col))
        else:
            non_exist_cols.append(node_col)
    if non_exist_cols:
        raise RuntimeError(
            ("The following input column(s) for an offloaded map operation "
             "do not exist: {}").format(non_exist_cols))

    return col_idxs