One_tone_power.py

import numpy as np
from matplotlib import pyplot as plt


#File path
directory = 'D:\Data\Fluxonium #28\One_tone'
measurement = '050818_Two_tone_power_YOKO_-30to5dBm_Cav_7.3367GHz&59.719mA_QuBit0.2to0.35GHz&25dBm'
path = directory + '\\' + measurement

#Read data
power = np.genfromtxt(path + '_POWER.csv')
power = power[1:-1]
freq = np.genfromtxt(path + '_FREQ.csv')
freq = freq[1::]
data = np.genfromtxt(path + '_PHASEMAG.csv')
phase = data[1::,0] #phase is recorded in rad
mag = data[1::,1]
magdB = 10*np.log10(mag)
Z = np.zeros((len(power), len(freq)))
# print (phase)
for idx in range(len(power)):
    # temp = np.unwrap(phase[idx*len(freq):(idx+1)*len(freq)])
    temp = magdB[idx*len(freq):(idx+1)*len(freq)]

    Z[idx,:] = temp - np.max(temp)

# Z = Z*180/(np.pi)

X,Y = np.meshgrid(power, freq)
plt.pcolormesh(X,Y,Z.transpose(), cmap= 'GnBu', vmin = 0, vmax=-10)


# plt.ylim([freq[0],freq[-1]])
plt.colorbar()
plt.show()