解析TMalign文本文件中的转换矩阵

TM-align 将两个蛋白质结构通过旋转和位移对齐后**:**

TMalign test1.pdb  test2.pdb -m mtx.txt

输出转换矩阵,文件内容为:

------ The rotation matrix to rotate Chain_1 to Chain_2 ------
m               t[m]        u[m][0]        u[m][1]        u[m][2]
0       0.7438770778   0.9928096071   0.1017845140   0.0629999746
1       3.6134235067   0.0559138503   0.0710345055  -0.9959054877
2      18.3578484222  -0.1058429281   0.9922671070   0.0648325754

Code for rotating Structure A from (x,y,z) to (X,Y,Z):
for(i=0; i<L; i++)
{
   X[i] = t[0] + u[0][0]*x[i] + u[0][1]*y[i] + u[0][2]*z[i];
   Y[i] = t[1] + u[1][0]*x[i] + u[1][1]*y[i] + u[1][2]*z[i];
   Z[i] = t[2] + u[2][0]*x[i] + u[2][1]*y[i] + u[2][2]*z[i];
}
  • 旋转矩阵 u 是一个 3x3 的矩阵,用于定义结构 A 到结构 B 的旋转。
  • 平移向量 t 是一个长度为 3 的向量,用于定义结构 A 到结构 B 的平移。

解析为numpy array代码

方法一:

import numpy as np

# 打开mtx.txt文本,读入数据
text = ""
with open("mtx.txt") as f:
    lines = f.readlines()

# 1. 取数据行
data_lines = lines[2:5]

# 初始化一个空列表来存储每一行的数值
data = []

# 遍历每一行,提取数值
for line in data_lines:
    # 将每行按空白字符分割,并将数值转换为 float
    values = [float(x) for x in line.split()[1:5]]
    data.append(values)

# 转换为 numpy 数组
array = np.array(data)

print(f"转换矩阵:{array}")

t = array[:,0] # 平移向量
u = array[:,1:] # 旋转矩阵
print(f"平移向量:{t}")
print(f"平移旋转:{u}")

方法二:

# 打开mtx.txt文本,读入数据
with open("mtx.txt") as f:
    lines = f.readlines()

# 1. 取数据行
data_lines = lines[2:5]

# 2. 去掉每行的第一个索引,留下数值部分
cleaned_lines = []
for line in data_lines:
    # 使用split分割并去掉每行的第一个元素(索引),保留数值部分
    cleaned_line = ' '.join(line.split()[1:5])
    cleaned_lines.append(cleaned_line)


print(f"cleaned_lines:{cleaned_lines}")

# 3. 将数值部分拼接成一个字符串,用于 fromstring 解析
data_string = ' '.join(cleaned_lines)
print("data_string")
print(data_string)

# 4. 使用 numpy.fromstring 解析数值字符串
array = np.fromstring(data_string, sep=' ')

# 5. 将生成的数组reshape为合适的形状 (3, 4)
array = array.reshape(3, 4)
print(array)

### 简洁形式
array = np.fromstring(' '.join(l[2:] for l in lines[2:5]), 
                  dtype=float, sep=' ').reshape((3,4))

print(f"转换矩阵:{array}")

t = array[:,0] # 平移向量
u = array[:,1:] # 旋转矩阵
print(f"平移向量:{t}")
print(f"平移旋转:{u}")
相关推荐
_.Switch18 分钟前
Python机器学习模型的部署与维护:版本管理、监控与更新策略
开发语言·人工智能·python·算法·机器学习
Hoper.J38 分钟前
PyTorch 模型保存与加载的三种常用方式
人工智能·pytorch·python
弱冠少年1 小时前
websockets库使用(基于Python)
开发语言·python·numpy
技术无疆2 小时前
【Python】Streamlit:为数据科学与机器学习打造的简易应用框架
开发语言·人工智能·python·深度学习·神经网络·机器学习·数据挖掘
羊小猪~~2 小时前
机器学习/数据分析--用通俗语言讲解时间序列自回归(AR)模型,并用其预测天气,拟合度98%+
人工智能·python·机器学习·数据挖掘·数据分析·回归·时序数据库
阿华的代码王国3 小时前
【JavaEE】——文件IO的应用
开发语言·python
电饭叔3 小时前
《python语言程序设计》2018版第8章19题几何Rectangle2D类(下)-头疼的几何和数学
开发语言·python
程序猿小D4 小时前
第二百六十七节 JPA教程 - JPA查询AND条件示例
java·开发语言·前端·数据库·windows·python·jpa
杰哥在此5 小时前
Python知识点:如何使用Multiprocessing进行并行任务管理
linux·开发语言·python·面试·编程