python 마스크 매트릭스 예제 구현(목록 에 따 른 요소)

2741 단어 pythonmask행렬
줄 과 열의 위 치 는 모두 아래 세 목록 의 한 열 에 있 으 며,대응 하 는 위 치 는 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 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]" + "
" + "[57-85,96-112,221-238]" + "
" + "[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 매트릭스 내용 은 우리 의 이전 글 을 검색 하거나 아래 의 관련 글 을 계속 찾 아 보 세 요.앞으로 많은 응원 바 랍 니 다!

좋은 웹페이지 즐겨찾기