这篇文章将为大家详细讲解有关怎么用Python实现流星雨效果,小编觉得挺实用的,因此分享给大家做个参考,希望大家阅读完这篇文章后可以有所收获。
流星雨的前提是得先有一个流星,所谓流星,就是一个拖着尾巴的直线。所谓拖着尾巴,实际上是我们的浪漫想象,实质无非是尺寸和颜色的渐变罢了。
而matplotlib
并不能给一条曲线映射多个颜色,故而可将一条线拆分成多条,效果如图所示
代码为
import numpy as npimport matplotlib.pyplot as pltfrom matplotlib.collections import LineCollectionx0,y0 = 1,1 #此为流星位置ts = np.arange(0,1,0.01) #参数xs,ys = x0+ts, y0+ts #绘图线条points = np.array([xs, ys]).T.reshape(-1, 1, 2)segments = np.concatenate([points[:-1], points[1:]], axis=1)ax = plt.subplot()lc = LineCollection(segments, cmap='viridis')lc.set_array(ts)lc.set_linewidth(ts[::-1])line = ax.add_collection(lc)ax.set_xlim(0, 3)ax.set_ylim(0, 3)plt.show()
接下来就要把这个流星变成流星雨,很简单,多加一些流星就可以了,效果如下,尽管是在白天,但是不是感觉有点意思了。
代码为
from numpy.random import rand, randintN,L = 20,100 #流星个数和线段数ts = np.array([ np.linspace(0,rand(),L) for _ in range(N)]).Tx0,y0 = rand(2*N).reshape(2,1,N)x0 *= 5xs,ys = x0+ts, y0+ts #绘图线条1points = np.array([xs, ys]).T.reshape(N,L,-1,2)ax = plt.subplot()for i in range(N): segs = np.concatenate([points[i][:-1], points[i][1:]], axis=1) lc = LineCollection(segs, cmap='viridis') lc.set_array(ts[:,i]) lc.set_linewidth(ts[::-1,i]) ax.add_collection(lc)ax.set_xlim(0, 6)ax.set_ylim(-2, 3)ax.set_axis_off() #取消坐标轴plt.show()
如果想让天黑下去,方法为
plt.figure(facecolor="black") #将背景设为黑色
此外,背景设为黑色之后,注意要把颜色映射调整一下,想知道更多具体的颜色映射表戳这里:matplotlib中的伪彩映射
lc = LineCollection(segs, cmap='viridis_r')
所以就是这种
关于“怎么用Python实现流星雨效果”这篇文章就分享到这里了,希望以上内容可以对大家有一定的帮助,使各位可以学到更多知识,如果觉得文章不错,请把它分享出去让更多的人看到。