Как построить с помощью matplotlib трехмерный график колчана с цветовым градиентом по длине? выдает ошибку "ValueError: слишком глубокий объект для нужного массива"

Я хочу указать длину колчанов (стрелы называют колчанами?), пометив их цветом. Это не проблема с 2D-графиками колчана. Вот готово. С 3D-проекцией плохо получается. Этот код воспроизводит проблему.

import matplotlib.pyplot as plt
import numpy as np

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

# Make the grid
x, y, z = np.meshgrid(np.arange(-0.8, 1, 0.2),
                      np.arange(-0.8, 1, 0.2),
                      np.arange(-0.8, 1, 0.8))

# Make the direction data for the arrows
u = np.sin(np.pi * x) * np.cos(np.pi * y) * np.cos(np.pi * z)
v = -np.cos(np.pi * x) * np.sin(np.pi * y) * np.cos(np.pi * z)
w = (np.sqrt(2.0 / 3.0) * np.cos(np.pi * x) * np.cos(np.pi * y) *
     np.sin(np.pi * z))
M = np.sqrt(u**2 + v**2 + w**2)
ax.quiver(x, y, z, u, v, w, M)

plt.show()

это создает длинную трассировку в нескольких файлах matplotlib и заканчивается

ValueError: object too deep for desired array

Как избежать этой проблемы и пометить колчаны цветом?