对于单点使用多种颜色的散点图

Question

希望在散点图中使用两种不同的颜色作为点。但是，这有效，第二个点仍然包含第一种颜色，并且仅由第二种颜色勾勒出来。见下面的例子。

用于生成此问题的代码：

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

swiss_roll, _ = make_swiss_roll(n_samples=100, noise=0.05, random_state=None)
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

#plot swiss roll
for xs, ys, zs in swiss_roll:
    ax.scatter(xs, ys, zs, color='b')

gs = np.random.uniform(-15 , 15, (100, 3))

for xs, ys, zs in gs:
    ax.scatter(xs, ys, zs, color='r')

ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')

plt.show()

我正在绘制单独的循环，因为中间有代码来生成gs。