操作数据数组和添加噪声

Question

我只是好像搞不好这件事！：-（

我有一个代码，它应该在.fits文件中读取，添加正态分布的噪声，然后重新保存该代码。到目前为止，它似乎根本不起作用。有很多额外的代码，所以我只发布了相关的部分。假设这段代码需要读入的所有内容都存在，因为它确实存在。此代码的目标是获取一个.fits文件，并向像素添加正态分布的噪声，然后保存该文件。”poisson”是先前输入的变量，即“poisson”值1对应于与零均值的一个标准偏差。是的，“poisson”这个词有点用词不当，我真的应该修改我的代码来修改它。在

我的第一个问题是。im0=im[0]是什么意思？它看起来不像是.fits文件中的第一行像素，因为当我将括号中的整数更改为除“0”之外的任何值时，就会出现索引错误。除此之外，法向噪声=np.随机.正态（0，poisson）方法不完整，因为我缺少第三个参数“size”（整数元组），我不知道这意味着什么。我的图像是130像素x 130像素，如果这意味着什么的话。在

    im = pf.open(name)
    im0 = im[0]
    normalNoise = np.random.normal(0,poisson)
    print im0.data
    test = im0.data + normalNoise
    print test
    im0.data = test
    stringee = 'NOISE'
    pf.writeto(stringee+str(poisson)+name, data=test, clobber=True, header=im0.header)
print poisson

这应该是理想的吐出相同的图像，但有额外的噪音，但它没有！在

Answer 1

我不知道底层库，但如果pf是FITS文件，im[0]可能是 创建对主HDU的引用；如果只有主HDU（对于 任何头脑简单的文件只适合一个图像），任何更高的索引都会导致错误。在