前提・実現したいこと
一次元配列x,yと、点(x,y)での色を表す一次元配列cを持っているとき、for文を使わずにプロットしたいです。
具体的には
python
1import numpy as np 2import matplotlib.cm as cm 3import matplotlib.pyplot as plt 4 5x = np.arange(0, 10,0.1) 6y = np.sin(x) 7 8norm_x = (x-min(x))/(max(x)-min(x)) 9c = cm.bone(norm_x) 10 11for i in range(len(x)): 12 plt.scatter(x[i],y[i],color=c[i]) 13 14plt.show()
を、
python
1import numpy as np 2import matplotlib.cm as cm 3import matplotlib.pyplot as plt 4 5x = np.arange(0, 10,0.1) 6y = np.sin(x) 7 8norm_x = (x-min(x))/(max(x)-min(x)) 9c = cm.bone(norm_x) 10 11plt.plot(x, y, 色=c) 12plt.show()
のような形に書き換えたいです。
調べたこと
plot_surfaceにはfacecolorsというオプションがあり、色を2次元配列で指定できるようです。
python
1from mpl_toolkits.mplot3d import Axes3D 2import matplotlib.pyplot as plt 3import numpy as np 4 5u = np.linspace(0, 2 * np.pi, 100) 6v = np.linspace(0, np.pi, 100) 7u, v = np.meshgrid(u,v) 8x = np.cos(u)*np.sin(v) 9y = np.sin(u)*np.sin(v) 10z = np.cos(v) 11z1 = z * np.cos(0.5*x) 12N = z1 / z1.max() 13 14fig = plt.figure() 15ax = fig.add_subplot(111, projection='3d') 16 17N = z1 / z1.max() 18ax.plot_surface(x, y, z, facecolors=cm.jet(N)) 19plt.show()
(ここから一部を変更して引用)
回答2件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。