MCPcopy Create free account
hub / github.com/dask/dask / plain_column_projection

Function plain_column_projection

dask/dataframe/dask_expr/_expr.py:3955–3972  ·  view source on GitHub ↗
(expr, parent, dependents, additional_columns=None)

Source from the content-addressed store, hash-verified

3953
3954
3955def plain_column_projection(expr, parent, dependents, additional_columns=None):
3956 column_union = determine_column_projection(
3957 expr, parent, dependents, additional_columns=additional_columns
3958 )
3959 if isinstance(column_union, list):
3960 column_union = [col for col in expr.frame.columns if col in column_union]
3961 elif column_union not in expr.frame.columns:
3962 # we are accessing the index
3963 column_union = []
3964
3965 if column_union == expr.frame.columns or not column_union and expr.ndim < 2:
3966 # this projection is for the index, but the elements are unknown, so
3967 # don't project
3968 return
3969 result = type(expr)(expr.frame[column_union], *expr.operands[1:])
3970 if column_union == parent.operand("columns"):
3971 return result
3972 return type(parent)(result, parent.operand("columns"))
3973
3974
3975def is_filter_pushdown_available(expr, parent, dependents, allow_reduction=True):

Callers 15

_simplify_upMethod · 0.90
_simplify_upMethod · 0.90
_simplify_upMethod · 0.90
_simplify_upMethod · 0.90
_simplify_upMethod · 0.90
_simplify_upMethod · 0.85
_simplify_upMethod · 0.85
_simplify_upMethod · 0.85
_simplify_upMethod · 0.85
_simplify_upMethod · 0.85
_simplify_upMethod · 0.85
_simplify_upMethod · 0.85

Calls 2

operandMethod · 0.80

Tested by

no test coverage detected