python實現mask矩陣示例(根據列表所給元素)

行和列的位置都在以下三個列表中的一列中，則對應位置為1，其余位置全為0

​ ——[7-56,239-327,438-454,522-556,574-586]

​ ——[57-85,96-112,221-238]

​ ——[113-220,328-437,455-521,557-573]

代碼實現

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位置數據 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 maskif __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實現mask矩陣示例(根據列表所給元素)的文章就介紹到這了,更多相關python實現mask矩陣 內容請搜索好吧啦網以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持好吧啦網！