實現(xiàn)功能

excel表格中有4列數(shù)，分別為RMF計算得到的 β，γ，勢能面及組態(tài)，需要挑選出相同 β 值下勢能面最低時的組態(tài)。為了減小數(shù)據(jù)量，先將 β 值保留兩位小數(shù)。

代碼

import xlrdimport xlwt# read xls filereadfile = xlrd.open_workbook(’./beta-gamma-constrain.xlsx’)readsheet = readfile.sheet_by_name(’Sheet1’)beta = readsheet.col_values(0)gamma = readsheet.col_values(1)energy = readsheet.col_values(2)config = readsheet.col_values(3)’’’print(beta)print(gamma)print(energy)print(config)’’’beta_2f = [round(x, 2) for x in beta]beta_gamma = dict(zip(beta_2f, gamma))beta_energy = dict(zip(beta_2f, energy))beta_config = dict(zip(beta_2f, config))for i in range(0, len(beta_2f)): if energy[i] < beta_energy[beta_2f[i]]: beta_gamma[beta_2f[i]] = gamma[i] beta_energy[beta_2f[i]] = energy[i] beta_config[beta_2f[i]] = config[i] else: continueprint(beta_gamma)print(beta_energy)print(beta_config)# write xls filewrite_excl = xlwt.Workbook(encoding=’utf-8’)excl_sheet = write_excl.add_sheet(’Sheet1’)j = 0for key, value in beta_gamma.items(): excl_sheet.write(j, 0, key) excl_sheet.write(j, 1, value) excl_sheet.write(j, 2, beta_energy[key]) excl_sheet.write(j, 3, beta_config[key]) j = j+1write_excl.save('xx.xls')

用到的庫

xlrd，讀取 excel 文件的庫，可以讀取 xls 和 xlsx 文件。

xlwt，寫入 excel 文件的庫，只能寫成 xls 文件。

思路

將數(shù)據(jù)按列讀出，寫入 4 個列表，再組裝為字典。由于字典中的 key 值是唯一的，因此該過程只是得到了 β-勢能面的字典，但勢能面的值不是最小的，需要遍歷判斷再賦值。最后將結(jié)果寫入新的 excel 表格。

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持好吧啦網(wǎng)。