corelogic/misc: contiguous split

This commit is contained in:
Sebastien Bourdeauducq 2012-02-11 11:52:15 +01:00
parent ef436a1ec9
commit e62ac1d3a1
1 changed files with 8 additions and 0 deletions

View File

@ -31,3 +31,11 @@ def optree(op, operands, lb=None, ub=None, default=None):
return _Operator(op,
[optree(op, operands, lb, s, default),
optree(op, operands, s, ub, default)])
def split(v, *counts):
r = []
offset = 0
for n in counts:
r.append(v[offset:offset+n])
offset += n
return tuple(r)