我总共产生50多个数字,但我不希望全部显示。我想挑选出我想要显示的数字。有没有办法做到这一点?
例如:我有50个在整个程序过程中生成的图。最后,当我输入plt.show()时,它会显示所有数字。但是,我想只显示3或4个数字(但它们不是固定的,即我可以一次绘制数字1,2,3,4,另一次我可以绘制数字10,27,33,45) 。此外,一个单独的功能正在生成这些数字,我将返回所有数字。
示例主脚本:
import matplotlib.pylab as plt
import numpy as np
from sampleplotfn import *
A = 1
omega = np.linspace(10,35,50)
for i in range(len(omega)):
fig1 = sinewave(A,omega[i])
plt.show()
samplepltfn.py
import numpy as np
import matplotlib.pylab as plt
def sinewave(A,omega):
t = np.linspace(0,10,25)
f = A*np.sin(omega*t)
fig1 = plt.figure()
plt.plot(f,t)
return fig1
答案 0 :(得分:8)
如果您真的只想绘制一些,只要您使用show
制作数字,就可以通过调用各个数字的plt.figure
方法来实现。例如。如果您制作了fig1和fig2,如果您拨打fig2.show()
而不是plt.show()
,则只能显示fig2。但是,我同意jeanrc的评论 - 你应该在制作它们之前清除你想要的情节,而不是保留大量未显示的情节(它们仍然占用记忆,你必须记得清理它们!)
此外,我强烈建议不要使用pylab,特别是反对使用“plt”作为它的导入别名。 “plt”传统上用于pyplot,而不是pylab。 (import matplotlib.pyplot as plt
)。 Pylab包含大量其他内容,除了交互式工作外,不鼓励使用。
最后,你在循环中覆盖“fig1”。尝试将数字保存到列表而不是单个变量中。
答案 1 :(得分:1)
根据您已经创建了2个图形,但是您只想显示第一个图形:
import random
f1, a1 = plt.subplots()
a1.plot(random.sample(range(100), 20))
f2, a2 = plt.subplots()
a2.plot(random.sample(range(100), 20))
f1.show()
在脚本中不起作用。
如评论中所建议,您可以在致电之前关闭不想看到的数字:
plt.close(f2)
plt.show()
一种适用于大量图形的更通用的解决方案是遍历所有图形并关闭它们,但要显示的图形除外:
for fig_num in plt.get_fignums():
if figure.number != fig_num:
plt.close(figure.number)
plt.show()
或者,使用生成器:
[plt.close(f) for f in plt.get_fignums() if f != f1.number]
plt.show()