python渐变图画法

Posted 2021-10-28 pocean2012

颜色变化的范例参见下文

python 画图colorbar 颜色大全 plt.cm.get_cmap_Lee_Yu_Rui的博客-CSDN博客

调用方式

先建立范例

 cm = plt.get_cmap("Greens_r")

填充数据

col = [cm(float(i)/(30)) for i in range(30)]

已打点图为例：

ax3D.scatter(x, y, z, s=10, c=col, marker='o')  

呈现效果

 

完整代码

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

# Create Map
cm = plt.get_cmap("Greens")

x = np.random.rand(30)
y = np.random.rand(30)
z = np.random.rand(30)
#col = [cm(float(i)/(29)) for i in xrange(29)] # BAD!!!
col = [cm(float(i)/(30)) for i in range(30)]

# 2D Plot
fig = plt.figure()
ax = fig.add_subplot(111)
ax.scatter(x, y, s=10, c=col, marker='o')  

# 3D Plot
fig = plt.figure()
ax3D = fig.add_subplot(111, projection='3d')
ax3D.scatter(x, y, z, s=10, c=col, marker='o')  

plt.show()