代码:
import numpy as np
import matplotlib.pyplot as plt
X, Y = (np.linspace(-3, 3, 100),
np.linspace(-3, 3, 100))
U, V = np.mgrid[-3:3:100j, 0:0:100j]
seed_points = np.array([[-2, 0, 1], [-2, 0, 1]])
fig0, ax0 = plt.subplots()
strm = ax0.streamplot(X, Y, U, V, color=U, linewidth=2,
cmap=plt.cm.autumn, start_points=seed_points.T)
fig0.colorbar(strm.lines)
ax0.plot(seed_points[0], seed_points[1], 'bo')
ax0.axis((-3, 3, -3, 3))
plt.show()
效果:
np.linspace:
1.1 功能
生成一个指定大小,指定数据区间的均匀分布序列。
1.2 参数说明
numpy.linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None)
(1)start:序列中数据的下界。
(2)end:序列中数据的上界。
(3)num:生成序列包含num个元素;其值默认为50。
(4)endpoint:取True时,序列包含最大值end;否则不包含;其值默认为True。
(5)retstep:该值取True时,生成的序列中显示间距;反正不显示;其值默认为false。
(6)dtype:数据类型,可以指定生成序列的数据类型;当为None时,根据其他输入推断数据类型。
(7)返回值:是一个数组。