4個Python庫來美化你的Matplotlib圖表!
來源:
奇酷教育 發(fā)表于:
4個Python庫來美化你的Matplotlib圖表!
4個Python庫來美化你的Matplotlib圖表!
而且每個方法都可以使用兩行代碼來實現(xiàn),一條import語句和一條Matplotlib的使用語句。
盡管這些庫非常適合創(chuàng)建酷炫的圖表,但是你還是需要了解你的目標受眾以及可能遇到的視覺問題,比如色盲等情況。
下面就讓我們來一起看看吧!
在開始使用探索Matplotlib圖表之前,先創(chuàng)建一些虛擬數(shù)據(jù),用來生成圖表。
幾行代碼輕松搞定。
import numpy as np
# 生成 x 值
x = np.linspace( 0 , 10 , 20 )
# 生成 y 值
y = np.sin(x)
y2 = np.cos(x)
01 賽博朋克風格——mplcyberpunk
賽博朋克是一種科幻小說的子類型,以其反烏托邦、技術(shù)先進的世界和反文化態(tài)度而聞名。
場景通常以未來主義風格描繪,經(jīng)常包括霓虹燈及明亮、鮮艷的色彩等元素。
比如,在制作海報或者是信息圖表時,你需要額外的元素來吸引讀者。這便是賽博朋克風格可以發(fā)揮作用的地方。
但是,你也要知道,它在為出版物創(chuàng)建圖表或在論文中使用時可能會被視為不專業(yè),并且可能不適合視覺有問題的讀者。
要開始使用這個主題,需要先安裝mplcyberpunk庫。
# 安裝
pip install mplcyberpunk
使用賽博朋克主題,只需要調(diào)用plt.style.use(),并傳入?yún)?shù)cypberpunk即可。
使用plt.scatter()繪制散點圖。為了使我們的散點發(fā)光,還需要調(diào)用make_scatter_glow()函數(shù)。
import mplcyberpunk
import matplotlib.pyplot as plt
# 使用賽博朋克風樣式
plt.style.use('cyberpunk')
plt.figure(figsize=(8, 8))
# 散點圖1
plt.scatter(x, y, marker='o')
mplcyberpunk.make_scatter_glow()
# 散點圖2
plt.scatter(x, y2, marker='o', c='lime')
mplcyberpunk.make_scatter_glow()
# 坐標軸名稱
plt.xlabel('X-Axis')
plt.ylabel('Y-Axis')
# 顯示
plt.show()
運行上面的代碼時,可以得到下圖。
將賽博朋克主題應用到折線圖中。為了使線條發(fā)光,我們可以調(diào)用make_lines_glow()。
# 使用賽博朋克風樣式
plt.style.use('cyberpunk')
plt.figure(figsize=(8, 8))
# 折線圖
plt.plot(x, y, marker='o')
plt.plot(x, y2, marker='o', c='lime')
# 發(fā)光效果
mplcyberpunk.make_lines_glow()
# 坐標軸名稱
plt.xlabel('X-Axis')
plt.ylabel('Y-Axis')
# 顯示
plt.show()
運行完成,將返回帶有霓虹燈線條的圖表。
我們還可以將圖表效果更進一步,在線條和X軸之間添加漸變填充。
# 使用賽博朋克風樣式
plt.style.use('cyberpunk')
plt.figure(figsize=(8, 8))
# 折線圖
plt.plot(x, y, marker='o')
plt.plot(x, y2, marker='o', c='lime')
# 發(fā)光效果+漸變填充
mplcyberpunk.add_gradient_fill(alpha_gradientglow=0.5, gradient_start='zero')
# 坐標軸名稱
plt.xlabel('X-Axis')
plt.ylabel('Y-Axis')
# 顯示
plt.show()
這便創(chuàng)造了一個非常有趣的效果。
當然這個賽博朋克庫還有其他的美化參數(shù),大家可以去GitHub上查看使用。
地址:https://github.com/dhaitz/mplcyberpunk
02 Matplotx
matplotx庫提供了一種簡單的方法可以快速對matplotlib圖表進行美化。
它包含幾個可以便捷使用,且可應用于任何matplotlib圖表的主題。
在深色主題的Jupyter Notebook或VSCode中處理圖表時,出現(xiàn)亮白色的圖表可能會有些許刺眼。
為了減少這種影響,我們可以將樣式設為深色主題。但是,這需要好幾行代碼才能實現(xiàn)。
正如接下來所看到的,Matplotx使這個過程變得更加容易。由于它包含多個主題,我們可以輕松找到一個與VSCode主題相匹配的。
通過打開終端/命令提示符并運行以下命令,可以將Matplotx安裝到你的Python環(huán)境中去。
# 安裝
pip install matplotx
安裝好以后,可以使用with語句來調(diào)用plt.style.context和傳入matplotx.styles。
在這里,我們可以選擇眾多的可用主題。 比如非常流行的Dracula主題。
import matplotx
with plt.style.context(matplotx.styles.dracula):
# 散點圖
plt.scatter(x, y, c=y2)
# 顏色類型
plt.colorbar(label='Y2')
# 坐標軸名稱
plt.xlabel('X')
plt.ylabel('Y')
# 顯示
plt.show()
運行代碼,得到下圖。
其中matplotx中有許多不同的樣式,具體情況如下。
下面讓我們用Pitaya Smoothie主題來創(chuàng)建一個折線圖。
由于有多個子主題,我們需要使用方括號訪問它們。
在這個例子中,我們有一個深色主題,所以我們需要傳入「dark」參數(shù)才能使用。
with plt.style.context(matplotx.styles.pitaya_smoothie['dark']):
# 折線圖
plt.plot(x, y, marker='o')
plt.plot(x, y2, marker='o')
# 坐標軸名稱
plt.xlabel('X')
plt.ylabel('Y')
# 顯示
plt.show()
當我們運行上面的代碼時,我們會得到下面這個折線圖。
當然也有許多淺色主題。比如Pitaya Smoothie有一個,可以像這樣使用。
with plt.style.context(matplotx.styles.pitaya_smoothie['light']):
如果你想了解更多有關(guān)此庫的信息,可以查看它的GitHub。
地址:https://github.com/nschloe/matplotx
03 量子黑色風格
QuantumBlack Labs是一家成立于2012年的公司,旨在幫助其他公司使用數(shù)據(jù)做出更好的決策。
他們使用機器學習和人工智能等先進技術(shù)來分析醫(yī)療保健、金融和交通等一系列行業(yè)的復雜數(shù)據(jù)集。
幾年前,他們在GitHub上發(fā)布了自己的樣式庫。
地址:
https://github.com/quantumblacklabs/qbstyles
要使用該主題,我們需要安裝這個Python庫。
# 安裝
pip install qbstyles
安裝好以后,就可以使用了~
from qbstyles import mpl_style
# 深色主題開
mpl_style(dark=True)
# 深色主題關(guān)
mpl_style(dark=False)
來創(chuàng)建一個散點圖,代碼如下。
from qbstyles import mpl_style
# 深色主題
mpl_style(dark=True)
# 非深色主題
mpl_style(dark=False)
# 散點圖
plt.scatter(x, y, c=y2)
# 顏色
plt.colorbar(label='Y2')
# 坐標軸名稱
plt.xlabel('X')
plt.ylabel('Y')
# 顯示
plt.show()
返回如下兩種圖表結(jié)果,具體取決于你選擇的是淺色還是深色主題。
讓我們看看深色主題的折線圖是什么樣子的。
# 深色主題
mpl_style(dark=True)
# 折線圖
plt.plot(x, y, marker='o')
plt.plot(x, y2, marker='o')
# 坐標軸名稱
plt.xlabel('X')
plt.ylabel('Y')
# 顯示
plt.show()
當我們運行上面的代碼時,我們會得到下面的圖。
這個庫的繪圖樣式與Matplotx中生成的繪圖相比略有緩和。
但是,它確實感覺更專業(yè)一些,尤其是淺色主題的繪圖。非常適合應用在專業(yè)演示或培訓課程材料中。
04 科學圖表
在撰寫科學期刊或會議文章時,創(chuàng)建一個清晰、簡單且易于解釋的圖表至關(guān)重要。
一些期刊,例如Nature,都是需要固定的樣式,方便簡化讀者難以理解的圖表。
這就是SciencePlots庫的用武之地。 它經(jīng)常被用于生成常見的科學期刊樣式圖表,從而使創(chuàng)建圖表變得更加容易。
這個庫的一大優(yōu)點是它使圖表適合黑白打印——這仍然是研究人員的普遍做法。通過更改線條樣式或確保分類數(shù)據(jù)的散點圖上有不同的形狀,這將使得線條很容易區(qū)分。
如果您想探索SciencePlots中更多可用的樣式,可以訪問它的GitHub。
地址:
https://github.com/garrettj403/SciencePlots/wiki/Gallery
運行SciencePlots庫需要在你的電腦上安裝LaTeX。你可以下面的地址找到有關(guān) LaTex以及如何安裝等更多詳細信息。
地址:https://www.latex-project.org/get/
在完成庫的安裝及LaTeX設置完成后,可以使用下面的代碼創(chuàng)建一個科學繪圖圖表。
import scienceplots
with plt.style.context(['science', 'high-vis']):
# 新建畫布
plt.figure(figsize=(6, 6))
# 折線圖
plt.plot(x, y, marker='o', label='Line 1')
plt.plot(x, y2, marker='o', label='Line 2')
# 坐標軸名稱
plt.xlabel('X')
plt.ylabel('Y')
# 圖例
plt.legend()
# 顯示
plt.show()
當我們運行上面的代碼時,我們得到以下圖表,這非常適合使用在期刊出版物中。
讓我們看看IEEE風格是什么樣的。我們可以通過修改參數(shù)來創(chuàng)建一個。
with plt.style.context(['science', 'ieee']):
和上面的有點不一樣,顏色也變了,但還是一個很好看的科學期刊圖表。
這對于搞學術(shù)研究的同學幫助非常大。
通過使用SciencePlots,不再為調(diào)整大小、顏色和線條樣式而煩惱。