python - Numpy 数组操作

Question

我有一个像 - 的数组 -

x =  array([0, 1, 2, 3,4,5])

我想要这样的输出 -

[]
[1]
[1 2]
[1 2 3]
[1 2 3 4]
[1 2 3 4 5]

我尝试了这段代码 -

y = np.array([np.arange(1,i) for i in x+1])

但它创建了一个包含我不想要的 dtype 对象的列表。我希望它不是整数，以便稍后可以对其进行索引。

Answer 1

如果我正确理解这个问题，是

y =  [np.arange(1,i) for i in x+1]

合适吗？您可以使用 y[r] 访问组成行的列表，例如

>>> y[2] 
array([1, 2])

或带有y的全部内容:

>>> y
[array([], dtype=int64),
 array([1]),
 array([1, 2]),
 array([1, 2, 3]),
 array([1, 2, 3, 4]),
 array([1, 2, 3, 4, 5])]

另请注意，您可以通过设置 dtype=int (或类似设置)来控制 arange 返回的数组的数据类型。