29 lines
722 B
Python
29 lines
722 B
Python
import time
|
|
import matplotlib.pyplot as plt
|
|
|
|
|
|
current_milli_time = lambda: int(round(time.time() * 1000))
|
|
|
|
|
|
def uniquefilename(name):
|
|
if '.' in name:
|
|
tmp = name.split('.')
|
|
return tmp[0] + str(current_milli_time()) + '.' + tmp[1]
|
|
else:
|
|
return name + str(current_milli_time())
|
|
|
|
|
|
def showAndSaveImage(fig,file,flag,lgd=None):
|
|
if flag:
|
|
plt.show()
|
|
if lgd is not None:
|
|
fig.savefig(uniquefilename(file), additional_artists=lgd,bbox_inches='tight') #bbox_extra_artists=(lgd,), )
|
|
else:
|
|
fig.savefig(uniquefilename(file))
|
|
plt.close(fig)
|
|
|
|
|
|
def enumerate2(xs, start=0, step=1):
|
|
for x in xs:
|
|
yield (start, x)
|
|
start += step |