def transform(block: Block) -> Block: block = BlockAccessor.for_block(block) builder = block.builder() for row in block.iter_rows(): if fn(row): builder.add(row) return builder.build()
def transform(block: Block) -> Block: block = BlockAccessor.for_block(block) builder = DelegatingArrowBlockBuilder() for row in block.iter_rows(): for r2 in fn(row): builder.add(r2) return builder.build()
def agg(block: Block) -> int: block = BlockAccessor.for_block(block) return sum(block.iter_rows())