Python: numpyのappendについて

Question

前提・実現したいこと

Pythonを使用しています。

• 実現したい内容 …

　ss[i].size==0のとき、ss[i-1][-1]を参照して
その要素を入れたいです。

個別に行うとうまくいくのですが、for文だとうまくいきません。

実行したコードとその結果

実行したコード

このようなコードを実行しました。

In[1]:
print(ss)

np.append(ss[2],ss[1])

In[2]:
for i in range(len(ss)-1):
    i += 1
    if ss[i].size == 0:
        np.append(ss[i],ss[i-1])
print(ss)

実行した結果

上記のコードを実行した結果、
以下のようになりました。

Out[1]:
array([array([[9.,  8.],[10.,  8.]]),
       array([[10.,  8.]]),
       array([], shape=(0, 2), dtype=float64),
       array([], shape=(0, 2), dtype=float64),
       array([], shape=(0, 2), dtype=float64),
       array([], shape=(0, 2), dtype=float64)], dtype=object)

array([10.,  8.])

Out[2]:
array([array([[9.,  8.],[10.,  8.]]),
       array([[10.,  8.]]),
       array([], shape=(0, 2), dtype=float64),
       array([], shape=(0, 2), dtype=float64),
       array([], shape=(0, 2), dtype=float64),
       array([], shape=(0, 2), dtype=float64)], dtype=object)

本来得たい結果

望んでいた実行結果は以下の通りです。

Out[2]:
array([array([[9.,  8.],[10.,  8.]]),
       array([[10.,  8.]]),
       array([[10.,  8.]]),
       array([[10.,  8.]]),
       array([[10.,  8.]]),
       array([[10.,  8.]])], dtype=object)

Comments

[meg_]

変数ssの元々の状態と求める計算結果がよくわかりません。

[TeRa_130]

ssは、

ss = np.array([np.array([[9,8],[10,8],]),np.array([[10,8]]),np.array([]),np.array([]),np.array([]),np.array([])])

となっています。

Answer 1

まず、np.appendはout-of-placeな関数なので、2番目のコードは「何もしない」のと同じです。1番目のコードもほぼそれに等しいのですが、Jupyter Notebookの仕様上最後に実行した式が表示されますので、一見できているように見えます。

そしてnumpy配列は作成したあとshapeを変えることが原則できません。今回はobject型の配列ですから、まあ代入してあげてください。

for i in range(1, ss.shape[0]):
    if ss[i].size == 0:
        ss[i] = ss[i-1][-1].reshape(1, -1)