如何在matplotlib中对3D散点图上的数据点进行着色

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

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

x = np.random.rand(100)
y = np.random.rand(100)
z = np.random.rand(100)
colors = np.random.rand(100)

ax.scatter(x, y, z, c=colors, cmap='viridis')

ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
plt.colorbar(ax.scatter(x, y, z, c=colors, cmap='viridis'), label='Color')

plt.show()