python 配列 reshape

Question

num_list =[]
for y in range(9):
  num_list.append(y)
  X = np.array([num_list])
  X = X.reshape(3, 3)
print(X)
---------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-54-f100f202c587> in <module>()
      2 for y in range(9):
      3   num_list.append(y)
----> 4   X = np.array([num_list]).reshape(3, 3)
      5 print(X)

ValueError: cannot reshape array of size 1 into shape (3,3)

とエラーが発生します。しかし

num_list =[]
for y in range(9):
  num_list.append(y)
  X = np.array([num_list])
print(X.reshape(3, 3))
#[[0 1 2]
  [3 4 5]
  [6 7 8]]

となります。
どうして一つ目のプログラムはエラーになるのでしょうか？
回答お願いします

Comments

[Yasumichi]

python はインデントが意味を持つので次のように書かないと正確なソースコードが、回答者に伝わりません。

```python
ここにコードを書く
```

詳しくは、以下も参考にしてください。
対応しているMarkdownの記法を知りたい
https://teratail.com/help#about-markdown

Answer 1

配列の形状変更：numpy.reshape, resize, newaxis より

引数は並び替え可能な数値を渡す必要があります。すなわち要素数は m で割り切れて、かつ変更後の配列のインデックスはすべて埋められなければなりません。要素数 11 の 1 次元配列を 3 × 4 の配列に変更しようとしても ValueEroor が発生します。

まず、最初のプログラムですと for のブロックの中で reshape(3, 3) を実行しているため、num_list の要素が９個になってなくても実行されてエラーとなっています。

以下のようにインデントを修正すれば、最初のプログラムでも動くはずです。

num_list =[]
for y in range(9):
  num_list.append(y)
X = np.array([num_list])
X = X.reshape(3, 3)
print(X)

※元が省略されてるので import は省略してます。

Comments

[barorin]

ありがとうございます

Answer 2

reshapeしている位置がfor文の外側か内側かの違いです。

1つめのコードでは、1回目のループでnum_list = [[0]]　になった後に3 × 3のリストにreshapeしようとしているため要素数が足りず実行時エラーになります。
2つ目のコードはnum_list = [0 1 2 3 4 5 6 7 8]と要素数が9になった後に3 × 3のリストにreshapeするためエラーになりません。

Comments

[退会済みユーザー]

回答作っている間に解決していたんですね^^;
失礼しました。