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

Method __setitem__

dask/dataframe/dask_expr/_collection.py:3051–3067  ·  view source on GitHub ↗
(self, key, value)

Source from the content-addressed store, hash-verified

3049 )
3050
3051 def __setitem__(self, key, value):
3052 if isinstance(key, (tuple, list)) and isinstance(value, DataFrame):
3053 out = self.assign(**{k: value[c] for k, c in zip(key, value.columns)})
3054
3055 elif isinstance(key, pd.Index) and not isinstance(value, DataFrame):
3056 out = self.assign(**dict.fromkeys(list(key), value))
3057 elif (
3058 is_dataframe_like(key)
3059 or is_series_like(key)
3060 or isinstance(key, (DataFrame, Series))
3061 ):
3062 out = self.where(~key, value)
3063 elif not isinstance(key, str):
3064 raise NotImplementedError(f"Item assignment with {type(key)} not supported")
3065 else:
3066 out = self.assign(**{key: value})
3067 self._expr = out._expr
3068
3069 def __delitem__(self, key):
3070 columns = [c for c in self.columns if c != key]

Callers

nothing calls this directly

Calls 4

assignMethod · 0.95
is_dataframe_likeFunction · 0.85
is_series_likeFunction · 0.85
whereMethod · 0.45

Tested by

no test coverage detected