QueryBuilder.select

from tmlt.analytics import QueryBuilder

QueryBuilder.select(columns)

Selects the specified columns, dropping the others.

Example

>>> my_private_data.toPandas()
   A  B  X
0  0  1  0
1  1  0  1
2  1  2  1
>>> budget = PureDPBudget(float("inf"))
>>> sess = Session.from_dataframe(
...     privacy_budget=budget,
...     source_id="my_private_data",
...     dataframe=my_private_data,
...     protected_change=AddOneRow(),
... )
>>> sess.get_column_types("my_private_data")
{'A': ColumnType.VARCHAR, 'B': ColumnType.INTEGER, 'X': ColumnType.INTEGER}
>>> # Create a new view using a select query
>>> query = (
...     QueryBuilder("my_private_data")
...     .select(["A", "B"])
... )
>>> sess.create_view(query, "selected_data", cache=True)
>>> # Inspect the schema of the resulting view
>>> sess.get_column_types("selected_data")
{'A': ColumnType.VARCHAR, 'B': ColumnType.INTEGER}

Parameters:

columns (Sequence[str]) – The columns to select.

Return type:

QueryBuilder