从matplotlib颜色栏中选取绘图颜色？

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 10, 100)
y = np.sin(x)

cmap = plt.cm.get_cmap('viridis')

plt.scatter(x, y, c=y, cmap=cmap)

plt.colorbar()

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 10, 100)
y = np.sin(x)

cmap = plt.cm.get_cmap('viridis')

plt.scatter(x, y, c=y, cmap=cmap)
plt.colorbar()

plt.show()