When erbium-doped fiber amplifier(EDFA) amplifies the signal of the same wavelength and different frequency, the signal distortion is caused by the different frequency response. In order to improve the communication quality, an EDFA frequency response test method is introduced, and a data processing system based on Python is written. The experimental results show that this method can complete the frequency response test task, process the data quickly and accurately, realize the data visualization, and improve the test efficiency.