Question

我正在尝试使用APLpy创建多个适合图像的子图，我希望能够通过for循环创建这些图，以防止我必须为N个图多次键入数十个参数。

使用不优雅的强力方法，对于N = 2的情节，它可能是这样的：

import aplpy
import matplotlib.pyplot as plt

fig = plt.figure()
f1 = aplpy.FITSFigure("figure1.fits", figure=fig, subplot=[0,0,0.5,0.5])
f2 = aplpy.FITSFigure("figure2.fits", figure=fig, subplot=[0.5,0.5,0.5,0.5])
# And there are many more images at this point, but let's look at 2 for now.

f1.show_colorscale()
f1.add_colorbar()
f1.frame.set_linewidth(0.75)
# And many more settings would follow

# Repeat this all again for the second plot
f2.show_colorscale()
f2.add_colorbar()
f2.frame.set_linewidth(0.75)

fig.canvas.draw()
fig.savefig('figure.eps')

但我想用for循环替换两组绘图参数，因为许多其他绘图参数都是以这种方式控制的，我想再做几个绘图。我想用以下内容替换这些行：

for i in range(1,3):
    f{i}.show_grayscale()
    f{i}.add_colorbar()
    f{i}.frame.set_linewidth(0.75)

等

显然这种语法是错误的。基本上我需要能够在for循环中修改代码本身。我在Python中找不到如何做到这一点，但如果我在.csh中做类似的事情，我可能会把它写成例如f"$i".show_grayscale()。

感谢。

Answer 1

今天我看到了解决这个问题的方法。 exec()命令允许您以这种方式执行一串代码。这种特殊情况的解决方案是使用：

for i in range(1,3):
    exec('f' + str(i) + '.show_grayscale()')
    exec('f' + str(i) + '.add_colorbar()')
    exec('f' + str(i) + '.frame.set_linewidth(0.75)')

这里的缺点是你在要执行的字符串中编写的代码没有通常具有的颜色编码格式。

Answer 2

一种方法是将FITSFigure对象添加到列表中：

fig = plt.figure(figsize=(8,10))
gc = []
gc.append(aplpy.FITSFigure(img1, subplot=[0.05,0.05,0.9,0.3], figure=fig))
gc.append(aplpy.FITSFigure(img2, subplot=[0.05,0.35,0.9,0.3], figure=fig))

然后你可以使用法线迭代：

for i in xrange(len(gc)):
  gc[i].recenter(ra, dec, radius=0.5)
  gc[i].tick_labels.hide()
  gc[i].axis_labels.hide()

使用for循环制作多个Python APLpy子图

2 个答案: