Pythonによる3Dグラフの描写ができない

Question

前提・実現したいこと

ラプラスの方程式の境界値問題について、反復法に基づく解法プログラムを作成し、3Dグラフを描写したいのですが、以下のエラーが発生してグラフが表示できません。
プログラムはオーム社「Pythonによる数値計算とシミュレーション」に掲載されたものをそのまま利用しています。

発生している問題・エラーメッセージ

Traceback (most recent call last):
  File "glaplace.py", line 42, in <module>
    ax.plot_wireframe(X, Y, u)
  File "/anaconda3/lib/python3.7/site-packages/mpl_toolkits/mplot3d/axes3d.py", line 1848, in plot_wireframe
    if Z.ndim != 2:
AttributeError: 'list' object has no attribute 'ndim'

該当のソースコード

# -*- coding: utf-8 -*-

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
import math

LIMIT = 1000    #反復回数の上限
N = 101         #x軸方向の分割数
M = 101         #y軸方向の分割数

#下請け関数の定義
#iteration()関数
def iteration(u) :

    u_next = [[0 for i in range(N)] for j in range(M)]

    for i in range(1, N-1) :
        for j in range(1, M-1) :
            u_next[i][j] = (u[i][j-1] + u[i-1][j] + u[i+1][j] + u[i][j+1]) / 4

    for i in range(1, N-1) :
        for j in range(1, M-1) :
            u[i][j] = u_next[i][j]
#iteration()関数の終わり

#メイン実行部
u = [[0 for i in range(N)] for j in range(M)]
for i in range(M) :
    u[0][i] = math.sin(2 * math.pi * i /(M-1))

for i in range(LIMIT) :
    iteration(u)

#グラフ描画
x = np.arange(0, N)
y = np.arange(0, M)
X, Y = np.meshgrid(x, y)
fig = plt.figure()
ax = Axes3D(fig)
ax.plot_wireframe(X, Y, u)
plt.show()

補足情報（FW/ツールのバージョンなど）

Python3.7

Answer 1

uはnumpy配列でないとだめなのでは？

安直に変換し、とりあえず実行できることを確認しました。

# ax.plot_wireframe(X, Y, u)  # 変更前
ax.plot_wireframe(X, Y, np.array(u))  # 変更後

Comments

[oldbrg425]

ax.plot_wireframe()について調べたところ、ご指摘の通り引数はnumpy配列でないといけないようでした。
ありがとうございました。

Answer 2

エラーで検索したらtopでヒットしました。

np.array 具体的な組み込み方は分かりませんが、これを使ってねってことかと。

model.fit expects x and y to be numpy array. Seems like you pass a list, it tried to get shape of input by reading ndim attribute of numpy array and failed.

You can simply transform it using np.array:

import numpy as np
...
model.fit(np.array(train_X),np.array(train_Y), epochs=20, batch_size=10)