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