小編這次要給大家分享的是python如何實現(xiàn)mask矩陣,文章內(nèi)容豐富,感興趣的小伙伴可以來了解一下,希望大家閱讀完這篇文章之后能夠有所收獲。

代碼實現(xiàn)
def generateMaskBasedOnDom(dom_path, length):
"""
:param dom_path: this is a file path, which contains the following information:
[7-56,239-327,438-454,522-556,574-586][57-85,96-112,221-238][113-220,328-437,455-521,557-573]
each [...] means one domain
:param length: this is the length of this protein
:return: the mask matrix with size length x length, 1 means inner domain residue pair, otherwise 0
"""
# 讀取文件
with open(dom_path, "r", encoding="utf-8") as file:
contents = file.readlines()
# 獲得mask位置數(shù)據(jù)
list0 = []
list1 = []
list2 = []
for list_idx, content in enumerate(contents):
num_range_list = content.strip()[1:-1].split(",")
for num_range in num_range_list:
start_num = int(num_range.split("-")[0])
end_num = int(num_range.split("-")[1])
for num in range(start_num, end_num+1):
if list_idx == 0:
list0.append(num)
elif list_idx == 1:
list1.append(num)
else:
list2.append(num)
mask = np.zeros((length, length))
# 遍歷矩陣每個元素
for row in range(mask.shape[0]):
for col in range(mask.shape[1]):
if (row in list0 and col in list0) or (row in list1 and col in list1) or (row in list2 and col in list2):
mask[row][col] = 1
return mask
if __name__ == "__main__":
# if no dom file ,please get dom file first
with open("dom.txt", "w", encoding="utf-8") as f:
f.write("[7-56,239-327,438-454,522-556,574-586]" + "\n" + "[57-85,96-112,221-238]" + "\n" + "[113-220,328-437,455-521,557-573]")
file_path = "./dom.txt"
protein_length = 1000 # mask_matrix size
mask_matrix = generateMaskBasedOnDom(file_path, protein_length)
print("*************Generate Mask Matrix Successful!*************")
# 隨機測試幾組
print(mask_matrix[7][56]) # 1
print(mask_matrix[7][239]) # 1
print(mask_matrix[8][57]) # 0
print(mask_matrix[57][95]) # 0
print(mask_matrix[113][573]) # 1
新聞標題:python如何實現(xiàn)mask矩陣-創(chuàng)新互聯(lián)
當前路徑:http://www.chinadenli.net/article40/djphho.html
成都網(wǎng)站建設公司_創(chuàng)新互聯(lián),為您提供面包屑導航、微信小程序、域名注冊、Google、用戶體驗、微信公眾號
聲明:本網(wǎng)站發(fā)布的內(nèi)容(圖片、視頻和文字)以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主,如果涉及侵權(quán)請盡快告知,我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場,如需處理請聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載,或轉(zhuǎn)載時需注明來源: 創(chuàng)新互聯(lián)
猜你還喜歡下面的內(nèi)容