Skip to content

enhanced_df

AbstractKlinkerFrame

Bases: ABC

Abstract klinker frame class

Source code in klinker/data/enhanced_df.py
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
class AbstractKlinkerFrame(ABC):
    """Abstract klinker frame class"""

    _table_name: Optional[str]
    _id_col: str

    @property
    def table_name(self) -> Optional[str]:
        """Name of dataset"""
        return self._table_name

    @table_name.setter
    def table_name(self, value: str):
        self._table_name = value

    @property
    def id_col(self) -> str:
        """Column where entity ids are stored"""
        return self._id_col

    @id_col.setter
    def id_col(self, value: str):
        self._id_col = value

    @property
    @abstractmethod
    def non_id_columns(self) -> List[str]:
        """Other columns than column with entity ids"""
        ...

    @abstractmethod
    def concat_values(
        self,
    ) -> SeriesType:
        """Concatenated entity attribute values.

        Returns:
            Concatenated attribute values as series with ids as index.
        """
        ...

    @classmethod
    @abstractmethod
    def _upgrade_from_series(
        cls,
        series: SeriesType,
        columns: List[str],
        table_name: Optional[str],
        id_col: str,
        reset_index: bool = True,
    ) -> "KlinkerFrame":
        """Upgrade series to KlinkerFrame.

        Args:
          series: SeriesType: series to upgrade
          columns: List[str]: column names of resulting df
          table_name: Optional[str]: dataset name
          id_col: str: name of id column
          reset_index: bool: whether to make id_col a seperate column

        Returns:
            klinker dataframe
        """
        ...

id_col: str property writable

Column where entity ids are stored

non_id_columns: List[str] abstractmethod property

Other columns than column with entity ids

table_name: Optional[str] property writable

Name of dataset

concat_values() abstractmethod

Concatenated entity attribute values.

Returns:

Type Description
SeriesType

Concatenated attribute values as series with ids as index.

Source code in klinker/data/enhanced_df.py
48
49
50
51
52
53
54
55
56
57
@abstractmethod
def concat_values(
    self,
) -> SeriesType:
    """Concatenated entity attribute values.

    Returns:
        Concatenated attribute values as series with ids as index.
    """
    ...

KlinkerDaskFrame

Bases: DataFrame, AbstractKlinkerFrame

Parallel KlinkerFrame.

Please don't use the __init__ method but rather from_dask_dataframe for initialisation!

Parameters:

Name Type Description Default
dsk

The dask graph to compute this KlinkerFrame

required
name

The key prefix that specifies which keys in the dask comprise this particular KlinkerFrame

required
meta

An empty klinkerframe object with names, dtypes, and indices matching the expected output.

required
divisions

Values along which we partition our blocks on the index

required

Returns:

Type Description

KlinkerDaskFrame

Examples:

>>> import pandas as pd
>>> from klinker.data import KlinkerDaskFrame
>>> import dask.dataframe as dd
>>> df = dd.from_pandas(pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"]),npartitions=1)
>>> kdf = KlinkerDaskFrame.from_dask_dataframe(df, table_name="A", id_col="id")
>>> kdf
Dask KlinkerDaskFrame Structure:
                   id first name surname
npartitions=1
0              object     object  object
1                 ...        ...     ...
Dask Name: KlinkerPandasFrame, 2 graph layers
Table Name: A, id_col: id
Source code in klinker/data/enhanced_df.py
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
class KlinkerDaskFrame(dd.core.DataFrame, AbstractKlinkerFrame):
    """Parallel KlinkerFrame.

    Please don't use the `__init__` method but rather `from_dask_dataframe` for
    initialisation!

    Args:
      dsk: The dask graph to compute this KlinkerFrame
      name: The key prefix that specifies which keys in the dask comprise this particular KlinkerFrame
      meta: An empty klinkerframe object with names, dtypes, and indices matching the expected output.
      divisions: Values along which we partition our blocks on the index

    Returns:
        KlinkerDaskFrame

    Examples:

        >>> import pandas as pd
        >>> from klinker.data import KlinkerDaskFrame
        >>> import dask.dataframe as dd
        >>> df = dd.from_pandas(pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"]),npartitions=1)
        >>> kdf = KlinkerDaskFrame.from_dask_dataframe(df, table_name="A", id_col="id")
        >>> kdf
        Dask KlinkerDaskFrame Structure:
                           id first name surname
        npartitions=1
        0              object     object  object
        1                 ...        ...     ...
        Dask Name: KlinkerPandasFrame, 2 graph layers
        Table Name: A, id_col: id

    """

    _partition_type = KlinkerPandasFrame

    def __init__(
        self,
        dsk,
        name,
        meta,
        divisions,
        table_name: Optional[str] = None,
        id_col: str = "id",
    ):
        super().__init__(dsk, name, meta, divisions)
        if table_name is None:
            self._table_name = meta.table_name
            self._id_col = meta.id_col
        else:
            self._table_name = table_name
            self._id_col = id_col

    @staticmethod
    def _static_propagate_klinker_attributes(
        new_object: "KlinkerDaskFrame", table_name: str, id_col: str
    ) -> "KlinkerDaskFrame":
        new_object.table_name = table_name
        new_object.id_col = id_col
        return new_object

    @property
    def non_id_columns(self) -> List[str]:
        """All columns which are not `id_col`"""
        return self._meta.non_id_columns

    @classmethod
    def _upgrade_from_series(
        cls,
        series,
        columns: List[str],
        table_name: Optional[str],
        id_col: str,
        reset_index: bool = True,
        meta=no_default,
    ) -> "KlinkerFrame":
        assert table_name
        kf = series.map_partitions(
            KlinkerPandasFrame._upgrade_from_series,
            columns=columns,
            table_name=table_name,
            id_col=id_col,
            reset_index=reset_index,
            meta=meta,
        )
        return KlinkerDaskFrame._static_propagate_klinker_attributes(
            kf, table_name, id_col
        )

    def concat_values(
        self,
    ) -> dd.Series:
        """Concatenate attribute values.

        Returns:
            dd.Series with concatenated values and id_col as index.

        Examples:

            >>> import pandas as pd
            >>> from klinker.data import KlinkerDaskFrame
            >>> import dask.dataframe as dd
            >>> df = dd.from_pandas(pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"]),npartitions=1)
            >>> kdf = KlinkerDaskFrame.from_dask_dataframe(df, table_name="A", id_col="id")
            >>> kdf.concat_values().compute()
            id
            1    John Doe
            2    Jane Doe
            Name: A, dtype: object

        """
        self = self.fillna("")
        assert self.table_name
        meta = pd.Series([], name=self.table_name, dtype="str")
        meta.index.name = self.id_col
        return self.map_partitions(
            M.concat_values,
            meta=meta,
        )

    @classmethod
    def from_dask_dataframe(
        cls,
        df: dd.DataFrame,
        table_name: str,
        id_col: str,
        meta=no_default,
        construction_class: Type[KlinkerPandasFrame] = KlinkerPandasFrame,
    ) -> "KlinkerDaskFrame":
        """Create KlinkDaskFrame from dask dataframe.

        Args:
          df: dd.DataFrame: Dask dataframe.
          table_name: str: Name of dataset.
          id_col: str: Column where entity_ids are stored
          meta: meta for dask
          construction_class: Either :class:`KlinkerPandasFrame` or :class:`KlinkerTriplePandasFrame`

        Returns:
            KlinkerDaskFrame

        Examples:

            >>> import pandas as pd
            >>> from klinker.data import KlinkerDaskFrame
            >>> import dask.dataframe as dd
            >>> df = dd.from_pandas(pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"]),npartitions=1)
            >>> kdf = KlinkerDaskFrame.from_dask_dataframe(df, table_name="A", id_col="id")
            >>> kdf
            Dask KlinkerDaskFrame Structure:
                               id first name surname
            npartitions=1
            0              object     object  object
            1                 ...        ...     ...
            Dask Name: KlinkerPandasFrame, 2 graph layers
            Table Name: A, id_col: id

        """
        new_df = df.map_partitions(
            construction_class,
            table_name=table_name,
            id_col=id_col,
            meta=meta,
        )
        meta = new_df._meta if meta is no_default else meta
        return cls(
            dsk=new_df.dask,
            name=new_df._name,
            meta=meta,
            divisions=new_df.divisions,
            table_name=table_name,
            id_col=id_col,
        )

    def __repr__(self) -> str:
        return (
            super().__repr__()
            + f"\nTable Name: {self.table_name}, id_col: {self.id_col}"
        )

non_id_columns: List[str] property

All columns which are not id_col

concat_values()

Concatenate attribute values.

Returns:

Type Description
Series

dd.Series with concatenated values and id_col as index.

Examples:

>>> import pandas as pd
>>> from klinker.data import KlinkerDaskFrame
>>> import dask.dataframe as dd
>>> df = dd.from_pandas(pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"]),npartitions=1)
>>> kdf = KlinkerDaskFrame.from_dask_dataframe(df, table_name="A", id_col="id")
>>> kdf.concat_values().compute()
id
1    John Doe
2    Jane Doe
Name: A, dtype: object
Source code in klinker/data/enhanced_df.py
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
def concat_values(
    self,
) -> dd.Series:
    """Concatenate attribute values.

    Returns:
        dd.Series with concatenated values and id_col as index.

    Examples:

        >>> import pandas as pd
        >>> from klinker.data import KlinkerDaskFrame
        >>> import dask.dataframe as dd
        >>> df = dd.from_pandas(pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"]),npartitions=1)
        >>> kdf = KlinkerDaskFrame.from_dask_dataframe(df, table_name="A", id_col="id")
        >>> kdf.concat_values().compute()
        id
        1    John Doe
        2    Jane Doe
        Name: A, dtype: object

    """
    self = self.fillna("")
    assert self.table_name
    meta = pd.Series([], name=self.table_name, dtype="str")
    meta.index.name = self.id_col
    return self.map_partitions(
        M.concat_values,
        meta=meta,
    )

from_dask_dataframe(df, table_name, id_col, meta=no_default, construction_class=KlinkerPandasFrame) classmethod

Create KlinkDaskFrame from dask dataframe.

Parameters:

Name Type Description Default
df DataFrame

dd.DataFrame: Dask dataframe.

required
table_name str

str: Name of dataset.

required
id_col str

str: Column where entity_ids are stored

required
meta

meta for dask

no_default
construction_class Type[KlinkerPandasFrame]

Either :class:KlinkerPandasFrame or :class:KlinkerTriplePandasFrame

KlinkerPandasFrame

Returns:

Type Description
KlinkerDaskFrame

KlinkerDaskFrame

Examples:

>>> import pandas as pd
>>> from klinker.data import KlinkerDaskFrame
>>> import dask.dataframe as dd
>>> df = dd.from_pandas(pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"]),npartitions=1)
>>> kdf = KlinkerDaskFrame.from_dask_dataframe(df, table_name="A", id_col="id")
>>> kdf
Dask KlinkerDaskFrame Structure:
                   id first name surname
npartitions=1
0              object     object  object
1                 ...        ...     ...
Dask Name: KlinkerPandasFrame, 2 graph layers
Table Name: A, id_col: id
Source code in klinker/data/enhanced_df.py
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
@classmethod
def from_dask_dataframe(
    cls,
    df: dd.DataFrame,
    table_name: str,
    id_col: str,
    meta=no_default,
    construction_class: Type[KlinkerPandasFrame] = KlinkerPandasFrame,
) -> "KlinkerDaskFrame":
    """Create KlinkDaskFrame from dask dataframe.

    Args:
      df: dd.DataFrame: Dask dataframe.
      table_name: str: Name of dataset.
      id_col: str: Column where entity_ids are stored
      meta: meta for dask
      construction_class: Either :class:`KlinkerPandasFrame` or :class:`KlinkerTriplePandasFrame`

    Returns:
        KlinkerDaskFrame

    Examples:

        >>> import pandas as pd
        >>> from klinker.data import KlinkerDaskFrame
        >>> import dask.dataframe as dd
        >>> df = dd.from_pandas(pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"]),npartitions=1)
        >>> kdf = KlinkerDaskFrame.from_dask_dataframe(df, table_name="A", id_col="id")
        >>> kdf
        Dask KlinkerDaskFrame Structure:
                           id first name surname
        npartitions=1
        0              object     object  object
        1                 ...        ...     ...
        Dask Name: KlinkerPandasFrame, 2 graph layers
        Table Name: A, id_col: id

    """
    new_df = df.map_partitions(
        construction_class,
        table_name=table_name,
        id_col=id_col,
        meta=meta,
    )
    meta = new_df._meta if meta is no_default else meta
    return cls(
        dsk=new_df.dask,
        name=new_df._name,
        meta=meta,
        divisions=new_df.divisions,
        table_name=table_name,
        id_col=id_col,
    )

KlinkerPandasFrame

Bases: DataFrame, AbstractKlinkerFrame

Enhanced pandas Dataframe for klinker.

This keeps table_name and id_col as metadata throughout transformations as best as possible.

Furthermore specific methods for blocking are implemented.

Examples:

>>> import pandas as pd
>>> from klinker.data import KlinkerPandasFrame
>>> df = pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"])
>>> df
  id first name surname
0  1       John     Doe
1  2       Jane     Doe
>>> kdf = KlinkerPandasFrame.from_df(df, table_name="A", id_col="id")
>>> kdf
  id first name surname
0  1       John     Doe
1  2       Jane     Doe
Table Name: A, id_col: id
>>> kdf.non_id_columns
['first name', 'surname']
>>> kdf.concat_values()
id
1    John Doe
2    Jane Doe
Name: A, dtype: object
Source code in klinker/data/enhanced_df.py
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
class KlinkerPandasFrame(pd.DataFrame, AbstractKlinkerFrame):
    """Enhanced pandas Dataframe for klinker.

    This keeps `table_name` and `id_col` as metadata
    throughout transformations as best as possible.

    Furthermore specific methods for blocking are implemented.

    Examples:

        >>> import pandas as pd
        >>> from klinker.data import KlinkerPandasFrame
        >>> df = pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"])
        >>> df
          id first name surname
        0  1       John     Doe
        1  2       Jane     Doe
        >>> kdf = KlinkerPandasFrame.from_df(df, table_name="A", id_col="id")
        >>> kdf
          id first name surname
        0  1       John     Doe
        1  2       Jane     Doe
        Table Name: A, id_col: id
        >>> kdf.non_id_columns
        ['first name', 'surname']
        >>> kdf.concat_values()
        id
        1    John Doe
        2    Jane Doe
        Name: A, dtype: object

    """

    _metadata = ["_table_name", "_id_col"]

    def __init__(
        self,
        data=None,
        index: Optional[Axes] = None,
        columns: Optional[Axes] = None,
        dtype: Optional[Dtype] = None,
        copy: Optional[bool] = None,
        table_name: Optional[str] = None,
        id_col: Optional[str] = "id",
    ) -> None:
        super().__init__(
            data=data, index=index, columns=columns, dtype=dtype, copy=copy
        )
        assert id_col
        self._table_name = table_name
        self._id_col: str = id_col

    @property
    def _constructor(self):
        """ """
        return KlinkerPandasFrame

    @property
    def non_id_columns(self) -> List[str]:
        """ """
        return [c for c in self.columns if not c == self.id_col]

    @classmethod
    def from_df(
        cls, df: pd.DataFrame, table_name: str, id_col: Optional[str] = "id"
    ) -> "KlinkerPandasFrame":
        """Construct a KlinkerPandasFrame from a pd.DataFrame.

        Args:
          df: pd.DataFrame: The df holding the data
          table_name: str: Name of the dataset.
          id_col: Optional[str]:  Column with entity ids ("id" as default).

        Returns:
            KlinkerPandasFrame

        Examples:

            >>> import pandas as pd
            >>> from klinker.data import KlinkerPandasFrame
            >>> df = pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"])
            >>> kdf = KlinkerPandasFrame.from_df(df, table_name="A", id_col="id")
            >>> kdf
              id first name surname
            0  1       John     Doe
            1  2       Jane     Doe
            Table Name: A, id_col: id

        """
        return cls(data=df, table_name=table_name, id_col=id_col)

    def concat_values(
        self,
    ) -> pd.Series:
        """Concatenate all values, that are not in the id_col.

        Returns:
            Series with id_col as index and concatenated values.

        Examples:

            >>> import pandas as pd
            >>> from klinker.data import KlinkerPandasFrame
            >>> df = pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"])
            >>> kdf = KlinkerPandasFrame.from_df(df, table_name="A", id_col="id")
            >>> kdf.concat_values()
            id
            1    John Doe
            2    Jane Doe
            Name: A, dtype: object

        """
        self = self.fillna("")
        result = (
            self.copy()
            .set_index(self.id_col)[self.non_id_columns]
            .astype(str)
            .agg(" ".join, axis=1)
            .str.strip()
        )
        result.name = self.table_name
        return result

    @classmethod
    def _upgrade_from_series(
        cls,
        series,
        columns: List[str],
        table_name: Optional[str],
        id_col: str,
        reset_index: bool = True,
    ) -> "KlinkerFrame":
        kf = KlinkerPandasFrame(series.to_frame(), table_name=table_name, id_col=id_col)
        if reset_index:
            kf = kf.reset_index()
        kf.columns = columns
        return kf

    def __repr__(self) -> str:
        return (
            super().__repr__()
            + f"\nTable Name: {self.table_name}, id_col: {self.id_col}"
        )

non_id_columns: List[str] property

concat_values()

Concatenate all values, that are not in the id_col.

Returns:

Type Description
Series

Series with id_col as index and concatenated values.

Examples:

>>> import pandas as pd
>>> from klinker.data import KlinkerPandasFrame
>>> df = pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"])
>>> kdf = KlinkerPandasFrame.from_df(df, table_name="A", id_col="id")
>>> kdf.concat_values()
id
1    John Doe
2    Jane Doe
Name: A, dtype: object
Source code in klinker/data/enhanced_df.py
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
def concat_values(
    self,
) -> pd.Series:
    """Concatenate all values, that are not in the id_col.

    Returns:
        Series with id_col as index and concatenated values.

    Examples:

        >>> import pandas as pd
        >>> from klinker.data import KlinkerPandasFrame
        >>> df = pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"])
        >>> kdf = KlinkerPandasFrame.from_df(df, table_name="A", id_col="id")
        >>> kdf.concat_values()
        id
        1    John Doe
        2    Jane Doe
        Name: A, dtype: object

    """
    self = self.fillna("")
    result = (
        self.copy()
        .set_index(self.id_col)[self.non_id_columns]
        .astype(str)
        .agg(" ".join, axis=1)
        .str.strip()
    )
    result.name = self.table_name
    return result

from_df(df, table_name, id_col='id') classmethod

Construct a KlinkerPandasFrame from a pd.DataFrame.

Parameters:

Name Type Description Default
df DataFrame

pd.DataFrame: The df holding the data

required
table_name str

str: Name of the dataset.

required
id_col Optional[str]

Optional[str]: Column with entity ids ("id" as default).

'id'

Returns:

Type Description
KlinkerPandasFrame

KlinkerPandasFrame

Examples:

>>> import pandas as pd
>>> from klinker.data import KlinkerPandasFrame
>>> df = pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"])
>>> kdf = KlinkerPandasFrame.from_df(df, table_name="A", id_col="id")
>>> kdf
  id first name surname
0  1       John     Doe
1  2       Jane     Doe
Table Name: A, id_col: id
Source code in klinker/data/enhanced_df.py
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
@classmethod
def from_df(
    cls, df: pd.DataFrame, table_name: str, id_col: Optional[str] = "id"
) -> "KlinkerPandasFrame":
    """Construct a KlinkerPandasFrame from a pd.DataFrame.

    Args:
      df: pd.DataFrame: The df holding the data
      table_name: str: Name of the dataset.
      id_col: Optional[str]:  Column with entity ids ("id" as default).

    Returns:
        KlinkerPandasFrame

    Examples:

        >>> import pandas as pd
        >>> from klinker.data import KlinkerPandasFrame
        >>> df = pd.DataFrame([("1","John", "Doe"),("2","Jane","Doe")],columns=["id","first name", "surname"])
        >>> kdf = KlinkerPandasFrame.from_df(df, table_name="A", id_col="id")
        >>> kdf
          id first name surname
        0  1       John     Doe
        1  2       Jane     Doe
        Table Name: A, id_col: id

    """
    return cls(data=df, table_name=table_name, id_col=id_col)

KlinkerTripleDaskFrame

Bases: KlinkerDaskFrame

Parallel KlinkerTriplePandasFrame

Parameters:

Name Type Description Default
dsk

The dask graph to compute this KlinkerFrame

required
name

The key prefix that specifies which keys in the dask comprise this particular KlinkerFrame

required
meta

An empty klinkerframe object with names, dtypes, and indices matching the expected output.

required
divisions

Values along which we partition our blocks on the index

required

Returns:

Source code in klinker/data/enhanced_df.py
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
class KlinkerTripleDaskFrame(KlinkerDaskFrame):
    """Parallel KlinkerTriplePandasFrame

    Args:
      dsk: The dask graph to compute this KlinkerFrame
      name: The key prefix that specifies which keys in the dask comprise this particular KlinkerFrame
      meta: An empty klinkerframe object with names, dtypes, and indices matching the expected output.
      divisions: Values along which we partition our blocks on the index

    Returns:

    """

    _partition_type = KlinkerTriplePandasFrame

    def concat_values(
        self,
    ) -> dd.Series:
        """


        Returns:

        """
        self = self.fillna("")
        assert self.table_name
        result = self.groupby(self.id_col)[self.columns[2]].apply(
            lambda grp: " ".join(grp.astype(str)).strip(),
            meta=pd.Series([], name=self.columns[2], dtype="str"),
        )
        result.name = self.table_name
        result._meta.index.name = self.id_col
        return result

concat_values()

Returns:

Source code in klinker/data/enhanced_df.py
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
def concat_values(
    self,
) -> dd.Series:
    """


    Returns:

    """
    self = self.fillna("")
    assert self.table_name
    result = self.groupby(self.id_col)[self.columns[2]].apply(
        lambda grp: " ".join(grp.astype(str)).strip(),
        meta=pd.Series([], name=self.columns[2], dtype="str"),
    )
    result.name = self.table_name
    result._meta.index.name = self.id_col
    return result

KlinkerTriplePandasFrame

Bases: KlinkerPandasFrame

Class for holding triple information.

Examples:

>>> import pandas as pd
>>> from klinker.data import KlinkerTriplePandasFrame
>>> df = pd.DataFrame([("e1","foaf:givenname","John"),("e1","foaf:family_name", "Doe"), ("e2","rdfs:label","Jane Doe")],columns=["head","rel","tail"])
>>> from klinker.data import KlinkerTriplePandasFrame
>>> kdf = KlinkerTriplePandasFrame.from_df(df, table_name="A",id_col="head")
>>> kdf
head               rel      tail
0   e1    foaf:givenname      John
1   e1  foaf:family_name       Doe
2   e2        rdfs:label  Jane Doe
Table Name: A, id_col: head
>>> kdf.concat_values()
head
e1    John Doe
e2    Jane Doe
Name: A, dtype: object
Source code in klinker/data/enhanced_df.py
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
class KlinkerTriplePandasFrame(KlinkerPandasFrame):
    """Class for holding triple information.

    Examples:

        >>> import pandas as pd
        >>> from klinker.data import KlinkerTriplePandasFrame
        >>> df = pd.DataFrame([("e1","foaf:givenname","John"),("e1","foaf:family_name", "Doe"), ("e2","rdfs:label","Jane Doe")],columns=["head","rel","tail"])
        >>> from klinker.data import KlinkerTriplePandasFrame
        >>> kdf = KlinkerTriplePandasFrame.from_df(df, table_name="A",id_col="head")
        >>> kdf
        head               rel      tail
        0   e1    foaf:givenname      John
        1   e1  foaf:family_name       Doe
        2   e2        rdfs:label  Jane Doe
        Table Name: A, id_col: head
        >>> kdf.concat_values()
        head
        e1    John Doe
        e2    Jane Doe
        Name: A, dtype: object

    """

    @property
    def _constructor(self):
        """ """
        return KlinkerTriplePandasFrame

    @property
    def non_id_columns(self) -> List[str]:
        """Last column."""
        return [self.columns[2]]

    def concat_values(
        self,
    ) -> pd.Series:
        """Concatenate all values of the tail column.

        Returns:
            Series with id_col as index and concatenated values.

        Examples:
            >>> import pandas as pd
            >>> from klinker.data import KlinkerTriplePandasFrame
            >>> df = pd.DataFrame([("e1","foaf:givenname","John"),("e1","foaf:family_name", "Doe"), ("e2","rdfs:label","Jane Doe")],columns=["head","rel","tail"])
            >>> from klinker.data import KlinkerTriplePandasFrame
            >>> kdf = KlinkerTriplePandasFrame.from_df(df, table_name="A",id_col="head")
            >>> kdf.concat_values()
            head
            e1    John Doe
            e2    Jane Doe
            Name: A, dtype: object
        """
        assert self.table_name
        self = self.fillna("")
        res = (
            self[[self.id_col, self.columns[2]]]
            .groupby(self.id_col)
            .agg(lambda row: " ".join(row.astype(str).values).strip())[self.columns[2]]
        )
        res.name = self.table_name
        return res

non_id_columns: List[str] property

Last column.

concat_values()

Concatenate all values of the tail column.

Returns:

Type Description
Series

Series with id_col as index and concatenated values.

Examples:

>>> import pandas as pd
>>> from klinker.data import KlinkerTriplePandasFrame
>>> df = pd.DataFrame([("e1","foaf:givenname","John"),("e1","foaf:family_name", "Doe"), ("e2","rdfs:label","Jane Doe")],columns=["head","rel","tail"])
>>> from klinker.data import KlinkerTriplePandasFrame
>>> kdf = KlinkerTriplePandasFrame.from_df(df, table_name="A",id_col="head")
>>> kdf.concat_values()
head
e1    John Doe
e2    Jane Doe
Name: A, dtype: object
Source code in klinker/data/enhanced_df.py
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
def concat_values(
    self,
) -> pd.Series:
    """Concatenate all values of the tail column.

    Returns:
        Series with id_col as index and concatenated values.

    Examples:
        >>> import pandas as pd
        >>> from klinker.data import KlinkerTriplePandasFrame
        >>> df = pd.DataFrame([("e1","foaf:givenname","John"),("e1","foaf:family_name", "Doe"), ("e2","rdfs:label","Jane Doe")],columns=["head","rel","tail"])
        >>> from klinker.data import KlinkerTriplePandasFrame
        >>> kdf = KlinkerTriplePandasFrame.from_df(df, table_name="A",id_col="head")
        >>> kdf.concat_values()
        head
        e1    John Doe
        e2    Jane Doe
        Name: A, dtype: object
    """
    assert self.table_name
    self = self.fillna("")
    res = (
        self[[self.id_col, self.columns[2]]]
        .groupby(self.id_col)
        .agg(lambda row: " ".join(row.astype(str).values).strip())[self.columns[2]]
    )
    res.name = self.table_name
    return res

from_klinker_frame(kf, npartitions)

Create KlinkerDaskFrame from KlinkerPandasFrame.

Parameters:

Name Type Description Default
kf KlinkerPandasFrame

KlinkerPandasFrame: Input dataframe

required
npartitions int

int: Number of partitions for dask.

required

Returns:

Type Description
KlinkerDaskFrame

KlinkerDaskFrame

Source code in klinker/data/enhanced_df.py
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
def from_klinker_frame(kf: KlinkerPandasFrame, npartitions: int) -> "KlinkerDaskFrame":
    """Create KlinkerDaskFrame from KlinkerPandasFrame.

    Args:
      kf: KlinkerPandasFrame: Input dataframe
      npartitions: int: Number of partitions for dask.

    Returns:
        KlinkerDaskFrame
    """
    if not kf.table_name:
        raise ValueError("KlinkerFrame needs to have a table_name set!")
    cls = (
        KlinkerTripleDaskFrame
        if isinstance(kf, KlinkerTriplePandasFrame)
        else KlinkerDaskFrame
    )
    return cls.from_dask_dataframe(
        dd.from_pandas(kf, npartitions=npartitions),
        table_name=kf.table_name,
        id_col=kf.id_col,
        meta=kf.head(0),
        construction_class=kf.__class__,
    )

generic_upgrade_from_series(conc, reset_index=False)

Upgrade a series to KlinkerFrame.

This automatically determines the correct KlinkerFrame class based on the given series class.

Note

This will use the series name as the resulting dataset name. The series index is assumed to be the entity ids.

Parameters:

Name Type Description Default
conc SeriesType

SeriesType: Series to upgrade.

required
reset_index bool

bool: If True resets index.

False

Returns:

Type Description
KlinkerFrame

KlinkerFrame

Examples:

>>> import pandas as pd
>>> from klinker.data import generic_upgrade_from_series
>>> ser = pd.Series(["John Doe","Jane Doe"],name="A",index=["e1","e2"])
>>> ser
e1    John Doe
e2    Jane Doe
Name: A, dtype: object
>>> generic_upgrade_from_series(ser, reset_index=True)
   id    values
0  e1  John Doe
1  e2  Jane Doe
Table Name: A, id_col: id
Source code in klinker/data/enhanced_df.py
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
def generic_upgrade_from_series(
    conc: SeriesType, reset_index: bool = False
) -> KlinkerFrame:
    """Upgrade a series to KlinkerFrame.

    This automatically determines the correct KlinkerFrame class
    based on the given series class.

    Note:
        This will use the series name as the resulting dataset name.
        The series index is assumed to be the entity ids.

    Args:
      conc: SeriesType: Series to upgrade.
      reset_index: bool: If True resets index.

    Returns:
        KlinkerFrame

    Examples:

        >>> import pandas as pd
        >>> from klinker.data import generic_upgrade_from_series
        >>> ser = pd.Series(["John Doe","Jane Doe"],name="A",index=["e1","e2"])
        >>> ser
        e1    John Doe
        e2    Jane Doe
        Name: A, dtype: object
        >>> generic_upgrade_from_series(ser, reset_index=True)
           id    values
        0  e1  John Doe
        1  e2  Jane Doe
        Table Name: A, id_col: id

    """
    frame_class: Type[KlinkerFrame]
    id_col = "id"
    if isinstance(conc, pd.Series):
        frame_class = KlinkerPandasFrame
        if conc.index.name is None:
            conc.index.name = "id"
        else:
            id_col = conc.index.name
    else:
        frame_class = KlinkerDaskFrame
        if conc.index.name is None:
            conc._meta.index.name = "id"
        else:
            id_col = conc.index.name
    columns = ["values"] if not reset_index else [id_col, "values"]
    return frame_class._upgrade_from_series(
        conc,
        columns=columns,
        table_name=conc.name,
        id_col=id_col,
        reset_index=reset_index,
    )