用Python實現(xiàn)流星雨效果的方法詳解
流星雨的前提是得先有一個流星,所謂流星,就是一個拖著尾巴的直線。所謂拖著尾巴,實際上是我們的浪漫想象,實質(zhì)無非是尺寸和顏色的漸變罷了。
而matplotlib并不能給一條曲線映射多個顏色,故而可將一條線拆分成多條,效果如圖所示

代碼為
import numpy as np import matplotlib.pyplot as plt from matplotlib.collections import LineCollection x0,y0 = 1,1 #此為流星位置 ts = np.arange(0,1,0.01) #參數(shù) 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, randint
N,L = 20,100 #流星個數(shù)和線段數(shù)
ts = np.array([
np.linspace(0,rand(),L) for _ in range(N)]).T
x0,y0 = rand(2*N).reshape(2,1,N)
x0 *= 5
xs,ys = x0+ts, y0+ts #繪圖線條1
points = 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") #將背景設(shè)為黑色
此外,背景設(shè)為黑色之后,注意要把顏色映射調(diào)整一下,想知道更多具體的顏色映射表戳這里:matplotlib中的偽彩映射
lc = LineCollection(segs, cmap='viridis_r')
所以就是這種

總結(jié)
本篇文章就到這里了,希望能夠給你帶來幫助,也希望您能夠多多關(guān)注腳本之家的更多內(nèi)容!
相關(guān)文章
python?manage.py?createsuperuser運行錯誤問題解決
這篇文章主要介紹了python?manage.py?createsuperuser運行錯誤,本文給大家分享錯誤復現(xiàn)及解決方案,感興趣的朋友一起看看吧2023-10-10
可用于監(jiān)控 mysql Master Slave 狀態(tài)的python代碼
用于監(jiān)控MySQL Master Slave 狀態(tài)的python代碼,有需要的朋友可以參考下2013-02-02
Python讀取文件內(nèi)容為字符串的方法(多種方法詳解)
這篇文章主要介紹了Python讀取文件內(nèi)容為字符串的方法,本文通過三種方式給大家介紹,在文章末尾給大家提到了python讀取txt文件中字符串,字符串用空格分隔的相關(guān)知識,需要的朋友可以參考下2020-03-03

