pandas读取写入文件

学python这么久了，还是第一次接触到pandas读写文件。
读csv文件，写xlsx文件：

# read data
path = 'C:/Users/Lenovo/Desktop/data.CSV'
data = pd.read_csv(path, sep=',') # ,keep_default_na=False) 有NaN的时候需要这个参数
data = pd.DataFrame(data,dtype=float).fillna(0.0) # 空值补为0 
x_test = np.array(data.iloc[:, [0]]) # 读取第一列，转换为数组，之后就直接使用
x_train = np.array(data.iloc[:, [1]]) 
y_train = np.array(data.iloc[:, [2]])

# write data
writer = pd.ExcelWriter('C:/Users/Lenovo/Desktop/cpb.xlsx')
y_data = pd.DataFrame(pr) # 这里pr是一个矩阵，一个元素写一列
y_data.to_excel(writer, 'Sheet1')
writer.save()
writer.close()