Python應(yīng)用實現(xiàn)處理excel數(shù)據(jù)過程解析
實現(xiàn)功能
excel表格中有4列數(shù),分別為RMF計算得到的 β,γ,勢能面及組態(tài),需要挑選出相同 β 值下勢能面最低時的組態(tài)。為了減小數(shù)據(jù)量,先將 β 值保留兩位小數(shù)。
代碼
import xlrd
import xlwt
# read xls file
readfile = 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:
continue
print(beta_gamma)
print(beta_energy)
print(beta_config)
# write xls file
write_excl = xlwt.Workbook(encoding='utf-8')
excl_sheet = write_excl.add_sheet('Sheet1')
j = 0
for 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+1
write_excl.save("xx.xls")
用到的庫
xlrd,讀取 excel 文件的庫,可以讀取 xls 和 xlsx 文件。
xlwt,寫入 excel 文件的庫,只能寫成 xls 文件。
思路
將數(shù)據(jù)按列讀出,寫入 4 個列表,再組裝為字典。由于字典中的 key 值是唯一的,因此該過程只是得到了 β-勢能面的字典,但勢能面的值不是最小的,需要遍歷判斷再賦值。最后將結(jié)果寫入新的 excel 表格。
以上就是本文的全部內(nèi)容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
python timestamp和datetime之間轉(zhuǎn)換詳解
這篇文章主要為大家詳細介紹了python timestamp和datetime之間轉(zhuǎn)換,具有一定的參考價值,感興趣的小伙伴們可以參考一下2017-12-12
解決python 執(zhí)行sql語句時所傳參數(shù)含有單引號的問題
這篇文章主要介紹了解決python 執(zhí)行sql語句時所傳參數(shù)含有單引號的問題,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06
Python Zipfile模塊進行ZIP文件的創(chuàng)建解壓信息獲取和加密等操作
這篇文章主要為大家介紹了Python Zipfile模塊進行ZIP文件的創(chuàng)建解壓信息獲取和加密等操作,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2024-01-01
使用Python圖像處理庫Pillow處理圖像文件的案例分析
本文將通過使用Python圖像處理庫Pillow,幫助大家進一步了解Python的基本概念:模塊、對象、方法和函數(shù)的使用,文中代碼講解的非常詳細,需要的朋友可以參考下2023-07-07
python使用html2text庫實現(xiàn)從HTML轉(zhuǎn)markdown的方法詳解
這篇文章主要介紹了python使用html2text庫實現(xiàn)從HTML轉(zhuǎn)markdown的方法,需要的朋友可以參考下2020-02-02
Python編程中用close()方法關(guān)閉文件的教程
這篇文章主要介紹了Python編程中用close()方法關(guān)閉文件的教程,是Python編程入門中的基礎(chǔ)知識,需要的朋友可以參考下2015-05-05

