STFT振动信号Python实践

使用Python中Scipy库对轴承振动信号做离散STFT

from scipy import signal
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
%matplotlib inline

path_all='D:\\15_07_10\\' 
data = pd.read_csv(path_all+"N15_M07_F10_" +'KI18'+ "_1.csv")
data1=data.values[:8192]
data2=data1.flatten()
time=np.arange(8192)
plt.figure(figsize=(20,10))
plt.plot(time,data2)
plt.show()

fs=64000 #信号采样频率
f, t, Zxx = signal.stft(data2, fs, nperseg=512)
plt.pcolormesh(t, f, np.abs(Zxx))
plt.title('STFT Magnitude')
plt.ylabel('Frequency [Hz]')
plt.xlabel('Time [sec]')
plt.show()

以上纯属自学练习，如有错误欢迎指正！