出力された各データを結合してデータフレームを作成したい

Question

前提・実現したいこと

FFT処理を行うプログラムを作成しています。
そこで、出力された周波数のデータとFFT処理をしたデータを結合してデータフレームを作成したいと思っています。

今回は、dataのように出力することを目指しています。
**（試したことの画像を参照してください。）
**

発生している問題・エラーメッセージ

該当のソースコード

import numpy as np
import pandas as pd
import os
import matplotlib.pyplot as plt
import openpyxl

file_name = input("Please input filename: ")

Sampling_frequency = int(input("Please input Sampling_frequency: "))

Number_of_iterations = int(input("Please input Number_of_iterations: "))

color_list=["black","gray","lightcoral","brown","darkred","red","tomato","sienna","sandybrown","darkorange","tan","gold","darkkhaki"
            ,"yellowgreen","olivedrab","darkturquoise","lightblue"
            ,"steelblue","cornflowerblue","navy","blue","indigo","darkviolet","purple","magenta","crimson","lightpink"]
for n in range(1,Number_of_iterations+1):
    data=pd.read_csv(os.path.join(r"Z:/lab/private/結果/5月/",file_name),header=None,usecols=list(range(1,n+1)),names=[f"{n}回目" for n in range(1,n+1)])
#データ数
N=len(data)
#サンプリング周波数
fs=Sampling_frequency
#窓関数
window=np.hamming(N)
#窓関数の補正値
acf=(1/(sum(window)/N))
#時間軸のデータ作成
t=np.arange(0,N*(1/fs),1/fs)
#周波数軸のデータ作成
fq=np.linspace(0,fs,N) #周波数軸　linspace(開始,終了,分割数)

fig = plt.figure(figsize=(6,4))


for n in range(1,Number_of_iterations+1):
    #2次元配列から1次元配列に変更
    data2=data[str(n)+"回目"].values
    #窓関数後の信号
    data_w=window*data2
    #読み込んだデータをフーリエ変換
    data_w_FFT=np.fft.fft(data_w)
    #FFTの複素数結果を絶対値に変換
    data_w_FFT_abs=abs(data_w_FFT)
    #窓補正
    Data=(data_w_FFT_abs)*acf
    plt.plot(fq[:int(N/2)+1],Data[:int(N/2)+1],color=color_list[n],label=str(n)+"回目")
    df=pd.DataFrame(Data[:int(N/2)+1],fq[:int(N/2)+1],columns=[str(n)+"回目"])
    Data_for_graph=df.head(84)
    print(df)
    print("最大値："+str(max(Data[:int(N/2)+1])))
ax.set_xlabel("frequency[Hz]",fontsize=14)
ax.set_ylabel("signal intensity",fontsize=14)
plt.xlim(0,2000)
ax = fig.add_subplot(111)
ax.legend(prop={"family":"MS Gothic"})
plt.show()

試したこと

現在の出力結果は以下のようになっています。（dfのように出力されます。）

補足情報（FW/ツールのバージョンなど）

Comments

[bsdfan]

FFTした結果の5回分をまとめてひとつのデータフレームにしたいということですか？

[keisokukunn]

はい。
周波数[fq]とFFTした結果の5回分を一つのデータフレームにしたいと思っています。

Answer 1

ループの中で作ったデータフレームを、リストに残していって、最後にconcatすればよいです。

dfs = []   ### ← 追加
for n in range(1,Number_of_iterations+1):
    # ...
    df=pd.DataFrame(Data[:int(N/2)+1],fq[:int(N/2)+1],columns=[str(n)+"回目"])
    dfs.append(df)   ### ← 追加
    # ...
    print("最大値："+str(max(Data[:int(N/2)+1])))
df = pd.concat(dfs, axis=1)   ### ← 追加