python3Dベクトルマップの色

Question

pythonの３次元ベクトルマップの各ベクトルの色をベクトルの長さに応じて、色をグラデーションでつけたいです。（cmapの"Spectral"などの色で。）
どのようにすればよろしいでしょうか。困っています。

現状のコードは下記の通りになります。
Xcoor,Ycoor,Zcoorがぷろっとする座標であり、Fx,Fy,Fzがプロットするベクトルの各座標成分になります。
#############################################################
from mpl_toolkits.mplot3d import axes3d
Gxy=510**13
Gz=51011
L_vector=np.sqrt(GxyFx[:,0]**2+GxyFy[:,0]2+GzFz[:,0]**2)
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.quiver(Xcoor109,Ycoor*109,Zcorr,GxyFx[:,0],GxyFy[:,0],GzFz[:,0],arrow_length_ratio=0.0, pivot='tip',colors="r")# #
plt.xlim([np.min(Xcoor109),np.max(Xcoor*109)])
plt.ylim([np.min(Ycoor10**9),np.max(Ycoor10**9)])
ax.set_zlim([0, 5])
plt.show()

Comments

[tiitoi]

質問欄コードは
```
コード
```
のように markdwon 記法を使ってください。
また、コードは一部ではなく、コピペすれば動かせる状態のものを記述してください

Answer 1

そのように使うためのハードルは高いです。

• matplotlib.cmの機能を把握して、自分で数値と色のデータ（RGBA形式）への変換を行うこと
• quiverの実装を知っていること

まず以下のページの議論を参照してみてください。

python - Adding colors to a 3d quiver plot in matplotlib - Stack Overflow

コードとしては以下のように書けば動きます（@matplotlib 3.0.3。将来のバージョンに渡って動作することは保証できません）。

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

np.random.seed(42)
data = np.random.normal(size=(10, 6))

fig = plt.figure()
ax = fig.gca(projection='3d')

# 各ベクトルの長さを計算する
L = np.linalg.norm(data[:,3:6] - data[:,0:3], axis=1)

# 色のデータを作る
cmap = cm.ScalarMappable(cmap=cm.Spectral)
colors = cmap.to_rgba(L)
## おまじないっぽいこと(後ろの項がないとうまくいかない。SO参照)
colors = list(colors) + [x for x in colors for _ in range(2)]

# LineCollectionのcolors引数として指定
ax.quiver(data[:,0], data[:,1], data[:,2],
          data[:,3], data[:,4], data[:,5],
          colors=colors)

plt.savefig("result.png")