带有Array输入的np.arange

Question

我希望在给定的长度值存储为数组的情况下同时创建多个范围。

示例：

lengths = np.array([1, 5, 10])

创建范围：

ranges = np.arange(lengths)
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

或者，我也可以使用np.linspace的解决方案，将一个步长数组作为num参数。

谢谢。

Answer 1

也许你只需要一个列表理解？

lengths = np.array([1, 5, 10])
ranges = [np.arange(length) for length in lengths]