从ifft信号中提取数据值

2024-10-16 17:25:23 发布

您现在位置:Python中文网/ 问答频道 /正文

如何从Fourier transformation中提取filtered signalvalues?Asifft返回一个复数,很难进行进一步计算

我的python代码:

import numpy as np

# Create a simple signal with two frequencies
dt = 0.001
t = np.arange(0,1,dt)
f = np.sin(2*np.pi*50*t) + np.sin(2*np.pi*120*t) # Sum of 2 frequencies
f_clean = f
noise = 2.5*np.random.randn(len(t))
f = f + noise              # Add some noise

## Compute the Fast Fourier Transform (FFT)

n = len(t)
fhat = np.fft.fft(f,n)                     # Compute the FFT
PSD = fhat * np.conj(fhat) / n             # Power spectrum (power per freq)
freq = (1/(dt*n)) * np.arange(n)           # Create x-axis of frequencies in Hz
L = np.arange(1,np.floor(n/2),dtype='int') # Only plot the first half of freqs

## Use the PSD to filter out noise
indices = PSD > 100       # Find all freqs with large power
PSDclean = PSD * indices  # Zero out all others
fhat = indices * fhat     # Zero out small Fourier coeffs. in Y
ffilt = np.fft.ifft(fhat) # Inverse FFT for filtered time signal

这里ffilt是返回complex number的滤波信号。我想用这个信号进行数学计算,但不确定提取值的过程


Tags: ofthefftsignalnpdtoutfourier
1条回答
网友
1楼 · 发布于 2024-10-16 17:25:23

下面您可以看到如何从复数中获取实值,特别是复数的实部和虚部,以及复数的大小

a = ffilt[10]
# (1.09200370931126+4.0997278010904346e-17j)

# Get real part    
a.real
# 1.09200370931126

# Get imaginary part
a.imag
# 4.0997278010904346e-17
    
# Get magnitude 
abs(a)
# 1.09200370931126 
# (the imaginary part is close to zero, so the magnitude 
# is almost equal to the real part)

您可以对整个阵列执行上述操作

ffilt.real
ffilt.imag
abs(ffilt)
    
# Bonus: phase
np.angle(ffilt)

我不知道你下一步的计算结果是什么,但你可能想使用震级(abs)

相关问题 更多 >