1 回答

TA貢獻(xiàn)1843條經(jīng)驗(yàn) 獲得超7個(gè)贊
在函數(shù)之外創(chuàng)建你的fig和ax,并ax作為一個(gè)帶有默認(rèn)值的參數(shù)傳遞(所以我們可以在函數(shù)中創(chuàng)建 ax 以保持向后兼容性:
def foo(data, ax=None):
if ax is None:
_, ax = plt.subplots()
這使您仍然可以foo(data)像以前一樣調(diào)用。
如果您需要根據(jù)要求多次繪制此“函數(shù)”,則:
fig, axes = plt.subplots(nrows=3)
foo(data1, axes[0])
foo(data2, axes[1])
foo(data3, axes[2])
如果fig還需要,請(qǐng)將其作為參數(shù)傳遞給函數(shù),或通過(guò)ax.figure. 代碼將變?yōu)椋?/p>
def foo(data, ax=None):
if ax is None:
fig, ax = plt.subplots()
else:
fig, ax = ax.figure, ax
添加回答
舉報(bào)