时域转频域再显示

import matplotlib.pyplot as plt

import numpy as np

import pandas as pd

import torch

import torch.fft as fft

df = pd.read_csv('train.csv')

df=df.drop(['ID'],axis=1)

nmp=df.to_numpy()

feature=nmp[:-20,:-1]

label=nmp[:-20,-1]#（210，240）

feature=torch.fft.fft(torch.Tensor(feature))

feature=torch.abs(feature)

feature=feature.detach().numpy()

ax1 = plt.subplot(2,1,1)

plt.sca(ax1)

plt.plot(range(len(nmp[1,:-1])),feature[1,:])

ax2 = plt.subplot(2,1,2)

plt.sca(ax2)

plt.plot(range(len(nmp[1,:-1])),nmp[1,:-1])

plt.show()