Method pop_from_dataframe

python/dgl/data/csv_dataset_base.py:119–125

(df: pd.DataFrame, item: str)

    @staticmethod
    def pop_from_dataframe(df: pd.DataFrame, item: str):
        ret = None
        try:
            ret = df.pop(item).to_numpy().squeeze()
        except KeyError:
            pass
        return ret


class NodeData(BaseData):