python实现mask矩阵示例(根据列表所给元素)
作者:troublemaker、 时间:2022-04-19 19:37:16
行和列的位置都在以下三个列表中的一列中,则对应位置为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]" + "\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
来源:https://blog.csdn.net/weixin_44912159/article/details/107395081
标签:python,mask,矩阵
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
利用python做表格数据处理
2021-07-15 10:39:08
![](https://img.aspxhome.com/file/2023/9/120889_0s.png)
SQL SERVER如何判断某个字段包含大写字母
2023-07-01 21:19:12
python子线程如何有序执行
2022-03-14 01:08:54
python 查看cpu的核数实现
2023-01-02 21:09:31
![](https://img.aspxhome.com/file/2023/3/80493_0s.png)
基于django 的orm中非主键自增的实现方式
2023-06-04 13:53:54
Python使用apscheduler模块设置定时任务的实现
2021-05-01 05:01:54
Python+pytorch实现天气识别
2021-09-23 18:49:13
![](https://img.aspxhome.com/file/2023/6/79566_0s.png)
Python实现向服务器请求压缩数据及解压缩数据的方法示例
2023-03-14 05:02:08
Python 递归式实现二叉树前序,中序,后序遍历
2022-09-22 17:38:32
![](https://img.aspxhome.com/file/2023/7/66777_0s.png)
python爬虫入门教程--HTML文本的解析库BeautifulSoup(四)
2023-08-12 23:45:10
pycharm安装包失败的解决方法
2022-03-29 04:24:04
![](https://img.aspxhome.com/file/2023/1/97281_0s.png)
python中readline判断文件读取结束的方法
2022-12-14 06:22:51
Tensorflow的DataSet的使用详解
2021-03-19 18:18:04
python字符串循环左移
2023-08-28 19:32:13
php让json_encode不自动转义斜杠“/”的方法
2023-08-19 17:04:28
Python流程控制 while循环实现解析
2023-02-07 04:16:33
![](https://img.aspxhome.com/file/2023/9/86179_0s.jpg)
python函数的两种嵌套方法使用
2022-01-14 08:06:58
python文件与路径管理方法
2022-06-15 13:38:06
python中requests使用代理proxies方法介绍
2023-04-28 09:00:33
python使用glob检索文件的操作
2022-04-26 19:05:49
![](https://img.aspxhome.com/file/2023/7/124697_0s.jpg)