循环实现：将格点数据绘制在等值线上

import numpy as np
import matplotlib.pyplot as plt


X = np.linspace(0.0, 5.0, 100)
Y = np.cos(2 * np.pi * X)

plt.figure()
plt.plot(X, Y, ‘-og‘, ms=3)


plt.show()

import numpy as np
import matplotlib.pyplot as plt


X = np.linspace(0.0, 5.0, 100)
Y = np.cos(2 * np.pi * X)

plt.figure()
plt.plot(X, Y, ‘-og‘, ms=3)

# ============================================================
# 根据y的值生成颜色，当然也可以根据X,Y生成的Z来生成color
colors = []
for y in Y:
    if y > 0.5:
        colors.append(‘red‘)
    elif y > 0.0:
        colors.append(‘yellow‘)
    elif y > -0.5:
        colors.append(‘green‘)
    else:
        colors.append(‘blue‘)

# 显示所有数字
for x, y, c in zip(X, Y, colors):
    s = ‘{:.2f}‘.format(y)
    plt.text(x-0.1, y-0.01, s, color=c, fontsize=10) # plt.text只能单个地显示文本，所以要用循环。
                                                     # 另：x, y都向左下偏移了一点
# ============================================================

plt.show()