带有numpy的范围数组
带有numpy的范围数组
提问于 2011-03-13 08:12:37
我有以下数组：
>>> x = numpy.array([2,4,2,3,1])
>>> x
array([2, 4, 2, 3, 1])我想要这些值的范围数组。我可以像这样创建它：
>>> numpy.hstack( (numpy.arange(v) for v in x) )
array([0, 1, 0, 1, 2, 3, 0, 1, 0, 1, 2, 0])在给定x的情况下，有没有一种更快的方法可以在不使用for循环的情况下使用numpy生成它？
回答 2
Stack Overflow用户
回答已采纳
发布于 2011-03-13 09:30:44
我想通了：
>>> x
array([2, 4, 2, 3, 1])
>>> ends = numpy.cumsum(x)
>>> ranges = numpy.arange(ends[-1])
>>> ranges = ranges - numpy.repeat(ends-x, x)
>>> ranges
array([0, 1, 0, 1, 2, 3, 0, 1, 0, 1, 2, 0])
>>> Stack Overflow用户
发布于 2018-01-12 02:16:32
这是不是真的更快？我也有类似的需求，而且
concatenate([range(l, r) for l, r in array((left, right)).T])
是两倍于
range(end[-1]) + repeat(left + end, right-left)
(其中end = cumsum(right - left)就像你的一样)。
(在我很短的经验中，repeat速度非常慢--至少在python 3.6中是如此)
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/5286531
复制相关文章
相似问题
腾讯云开发者
