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])]

另请注意，您可以arange通过设置dtype=int（或类似方法）来控制此处返回的数组的数据类型。

Answer 2

我想要这样的输出

以这种方式输出它只是一种切片：

import numpy as np
x =  np.array([0, 1, 2, 3, 4, 5])

for i in range(1,len(x) + 1):
    print(x[1:i])