成像模拟切片并添加形状

时间:2016-06-07 11:14:11

标签: python data-visualization physics astronomy

我一直在制作模拟的切片图像,现在我需要为图像添加一个形状,切片有一个颜色贴图,我在切片上添加一个圆圈,我需要帮助使圆形颜色可以通过值,并与切片共享相同的颜色图。我使用的代码是:

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import csv

 def readslice(ndim):
    shape = (ndim,ndim,ndim)
    fd = open("delta_T_v3_no_halos_z013.40_nf0.898994_useTs0_zetaX-1.0e+00_alphaX-1.0_TvirminX-1.0e+00_aveTb027.03_Pop-1_300_200Mpc", 'rb')
    data = np.fromfile(file=fd, dtype= np.dtype('f4')).reshape(shape)
    fd.close()
    print data
    return data
ff =  readslice(300)

circle1=plt.Circle((150.0,150.0),50.0)
fig = plt.gcf()
fig.gca().add_artist(circle1)

plt.imshow(ff[0,:,:],cmap = cm.jet)
plt.colorbar()

plt.savefig('picwithcircle.png')
plt.show()

0 个答案:

没有答案