Class RandomChoice

dask/array/_array_expr/random.py:993–1027

class RandomChoice(IO):
    _parameters = [
        "array",
        "chunks",
        "_meta",
        "_state",
        "replace",
        "p",
        "axis",
        "shuffle",
    ]
    _defaults = {"axis": None, "shuffle": None}
    _funcname = "da.random.choice-"

    @cached_property
    def chunks(self):
        return self.operand("chunks")

    @cached_property
    def sizes(self):
        return list(product(*self.chunks))

    @cached_property
    def state_data(self):
        return random_state_data(len(self.sizes), self._state)

    @cached_property
    def _meta(self):
        return self.operands("_meta")

    def _layer(self) -> dict:
        return {
            k: (_choice_rs, state, self.array, size, self.replace, self.p)
            for k, state, size in zip(self.__dask_keys__(), self.state_data, self.sizes)
        }


class RandomChoiceGenerator(RandomChoice):