我想显示图像的幅度谱。我可以使用以下代码来做到这一点:
import numpy as np
import matplotlib.pyplot as plt
import pylab
pylab.gray()
pic = pylab.imread("C:/pic.png")[::-1,:]
amp_pic = pylab.subplot(1,4,1)
amp_pic.xaxis.set_ticks_position('top')
pylab.imshow(np.abs(np.fft.fftshift(np.fft.fft2(pic))),\
interpolation='nearest')
pylab.show()
但是轴的标记方式与幅度谱的标记方式不同。在一维函数的情况下,重新标记要容易一些:
import math
import numpy as np
import matplotlib.pyplot as plt
array = np.arange(149)
frequency_scale = np.fft.fftfreq(array.shape[0])
function = np.cos(2*math.pi*array/10)
fft = np.fft.fft(function)
amp_fft = np.abs(fft)
plt.subplot(1,4,1)
plt.plot(fkt,'r')
plt.show()
在 imshow 图的情况下,我希望我的 xaxis 具有相同的标签。这可能吗?