用python处理excel 数据分析_Python应用实现处理excel数据过程解析
實現功能
excel表格中有4列數,分別為RMF計算得到的 β,γ,勢能面及組態,需要挑選出相同 β 值下勢能面最低時的組態。為了減小數據量,先將 β 值保留兩位小數。
代碼
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 文件。
思路
將數據按列讀出,寫入 4 個列表,再組裝為字典。由于字典中的 key 值是唯一的,因此該過程只是得到了 β-勢能面的字典,但勢能面的值不是最小的,需要遍歷判斷再賦值。最后將結果寫入新的 excel 表格。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持python博客。
總結
以上是生活随笔為你收集整理的用python处理excel 数据分析_Python应用实现处理excel数据过程解析的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: word2vec相似度计算_文档相似度助
- 下一篇: python中类的定义方法_在Pytho