python计算海水碳酸盐体系

import numpy as np
import PyCO2SYS as pyco2
import sys
import os
f = open(r"C:\Users\huain\Desktop\c.txt")
line = f.readline()
data_list = []
while line:
    num = list(map(float,line.split()))
    data_list.append(num)
    line = f.readline()
f.close()
data_array = np.array(data_list)

pyco2_kws = dict(
    par1 = data_array[2],  # dissolved inorganic carbon in μmol/kg
    par2 = data_array[1],  # pH on the Total scale
    par1_type = 1,
    par2_type = 3,
    salinity = data_array[3],
    temperature = data_array[0],  # Temperature at input conditions
    opt_k_carbonic = 4,  # Choice of H2CO3 and HCO3- dissociation constants K1 and K2 ("4" means "Mehrbach refit")
    opt_k_bisulfate = 1,  # Choice of HSO4- dissociation constants KSO4 ("1" means "Dickson")
)
# Run PyCO2SYS and print out the pCO2 values
results = pyco2.sys(**pyco2_kws)
print("pCO2: ", results["pCO2"])
#将要输出保存的文件地址，若文件不存在，则会自动创建
fw = open(r"C:/Users/huain/Desktop\test.txt", 'w') 
#这里平时print("test")换成下面这行，就可以输出到文本中了
fw.write('results = pyco2.sys(**pyco2_kws)')  
# 换行
fw.write("\n")