情節大量通過循環

Question

說我有一堆AX1，AX2，AX3的使用一個命令軸的對象......我想通過一個繪圖功能來運行它們：

def plotxy(ax,x,y): 
    x = np.array(x) 
    y = np.array(y) 
    ax.plot(x,y) 

（明顯簡化）

我怎麼會重複該命令而不做：

plotxy(ax1,x,y) 
plotxy(ax2,x,y) 
... 
plotxy(axn,x,y) 

我敢肯定有一種方法來創建保持在一個循環中AXN這可能只是一條線一個臨時變量。有任何想法嗎？這會大大縮短我的代碼，因爲我有許多事情需要使用相同的命令進行繪製，但是不同的x和y多個子圖都是不同的。

我想這是使用循環構造變量名稱的一個大問題的一部分？

Answer 1

你可以嘗試這樣的事：如果你使用plt.subplot或plt.axes您可以創建一個列表

import matplotlib.pyplot as plt 
fig, axs = plt.subplots(nrows=3, ncols=2) 

for ax in axs.flat: 
    plotxy(ax,x,y) 

/手動軸列

Answer 2

我可能在你的問題中遺漏了一些東西，但你爲什麼不用一個簡單的循環？

for ax in (ax1, ax2, ..., axn): 
    plotxy(ax,x,y) 

您也可避免明確的列出所有的變量：

axes = (locals()['ax%d'%i] for i in range(1, n+1)) 
for ax in axes: 
    plotxy(ax,x,y) 

Answer 3

您還可以使用map，x和y是名字空間的變量和axx一些迭代在你的對象軸：

x = ... 
y = ... 

def plotxy(ax): 
    ax.plot(x,y) 

map(plotxy,axx)