pythonファイルを別のpythonプログラムから実行したい。

Question

USB_1.pyというpythonファイルを外部の別のpythonプログラムから実行し、その結果を得たいのですがうまくいきません。

import serial
from time import sleep
import subprocess
import pandas as pd
import numpy as np
import time
import sys
import csv
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt

df1 = pd.DataFrame(columns=['right'])
df2 = pd.DataFrame(columns=['left'])
N=1024 #data
dt=0.03125 #sample
t=np.arange(0,N*dt,dt)
freq=np.linspace(0,32/dt,N)


def main():
    ArdName={'Right_Hand_1':None,'Left_Hand_11':None}
    UsbSerial = [0 for i in range(3)]
    USBinfo = subprocess.getoutput('ls /dev | grep ttyACM*')
    USBinfo = USBinfo.split('\n')
    print(USBinfo)


    for i in range(len(USBinfo)):
        print(USBinfo[i])
        UsbSerial[i] = serial.Serial("/dev/%s"%USBinfo[i],115200)
        print(UsbSerial[i])
    sleep(2)

    for i in range(len(USBinfo)):
        UsbSerial[i].write('#Tell_Me_Your_Name#'.encode())
        x = {UsbSerial[i].read(12)}

    sleep(2)


    for i in range(len(USBinfo)):
        ArdName[subprocess.getoutput('cat /dev/%s'%USBinfo[i])]=UsbSerial[i]
        UsbSerial[i].write('#ok#'.encode())

    for loop in range(1024):
        data1 = UsbSerial[0].readline()
        data2 = UsbSerial[1].readline()
        data1=data1.decode()
        data2=data2.decode()
        data1=data1.rstrip()
        data2=data2.rstrip()
        data1 = int(data1)
        data2 = int(data2)
        global df1
        global df2
        df1=df1.append({'right':data1},ignore_index=True)
        df2=df2.append({'left':data2},ignore_index=True)

    UsbSerial[i].close()
    df1.to_csv('myakuha_right')
    df2.to_csv('myakuha_left')
#Serial open
    ser =serial.Serial('/dev/ttyUSB0',9600)
#Right array flat
    array2 = np.array(df1)
    array1=array2.flatten()
#Left array flat
    array3=np.array(df2)
    array4=array3.flatten()

    #print(array1)#right
    #print(array4)#left
#matplot
    plt.figure(figsize=(8,4))
    plt.plot(array1)
    plt.plot(array4)
    plt.savefig('myakuha.png')
    plt.show()
#ndarry FFT
    FFT=np.fft.fft(array1)
    FFT2=np.fft.fft(array4)



if __name__ == '__main__':
    main()

上記プログラムを以下のプログラムから実行し、FFTの結果を得要としましたが、エラーが出ました。

import USB_1

print(USB_1.FFT)

以下がエラーです。

Traceback (most recent call last):
  File "/home/pi/ttt.py", line 3, in <module>
    print(USB_1.FFT)
AttributeError: 'module' object has no attribute 'FFT'

解決策を教えていただけると嬉しいです。

Answer 1

変数FFTは、USB_1.pyの中のmain関数の中にあるローカル変数です。そのため、外部からは参照できません。
例えば、main関数を変えてFFTの値を返すようにしたり、USB_1.py内でFFT変数をグローバル変数としてそこに値を格納するなどの手がありえます。

Answer 2

main関数の末尾にreturn FFT
と付け足して、
print(USB_1.FFT)
の代わりに、

FFT = USB_1.main()
print(FFT)

としたら、行けると思います。