使用Python进行序列分析：基于状态转移矩阵的PCA分析

本代码演示了如何使用Python进行序列分析，并基于状态转移矩阵进行PCA分析。代码使用了Biopython库读取fasta文件，并利用numpy和matplotlib库进行数据处理和绘图。

from Bio import SeqIO
import numpy as np
import matplotlib.pyplot as plt
import os

# 读取fasta文件，计算状态转移矩阵
def calc_transition_matrix(fasta_file):
    sequences = list(SeqIO.parse(fasta_file, 'fasta'))
    alphabet = list(set(''.join([str(seq.seq) for seq in sequences])))  # 使用单引号
    states = len(alphabet)
    matrix = np.zeros((states, states))
    for seq in sequences:
        seq_str = str(seq.seq)
        for i in range(len(seq_str) - 1):
            from_state = alphabet.index(seq_str[i])
            to_state = alphabet.index(seq_str[i + 1])
            matrix[from_state, to_state] += 1
    return matrix

# 对矩阵进行z-score标准化
def standardize(matrix):
    standardized_matrix = (matrix - np.mean(matrix, axis=0)) / np.std(matrix, axis=0)
    return standardized_matrix

# 对标准化后的矩阵进行PCA分析
def pca(matrix):
    cov = np.cov(matrix.T)
    eig_vals, eig_vecs = np.linalg.eig(cov)
    idx = np.argsort(eig_vals)[::-1]
    eig_vecs = eig_vecs[:, idx]
    projection = np.dot(matrix, eig_vecs)
    return projection

# 测试代码
if __name__ == "__main__":
    # 获取文件夹中的所有fasta文件
    fasta_folder = "fasta_files"
    fasta_files = [os.path.join(fasta_folder, f) for f in os.listdir(fasta_folder) if f.endswith(".fasta")]

    # 对每个fasta文件进行状态转移矩阵的计算、标准化和PCA分析
    for fasta_file in fasta_files:
        # 读取fasta文件，计算状态转移矩阵
        matrix = calc_transition_matrix(fasta_file)
        print(matrix)

        # 对矩阵进行标准化
        standardized_matrix = standardize(matrix)
        print(standardized_matrix)

        # 对标准化后的矩阵进行PCA分析
        pca_result = pca(standardized_matrix)

        # 绘制PCA散点图
        plt.scatter(pca_result[:, 0], pca_result[:, 1], label=fasta_file)
    
    plt.legend()
    plt.show()

如何更改matplotlib的字体设置

可以通过以下步骤更改matplotlib的字体设置：

找到matplotlib的配置文件matplotlibrc。可以通过以下代码找到该文件的路径：

import matplotlib
print(matplotlib.matplotlib_fname())

打开matplotlibrc文件，找到以下两行：

#font.family         : sans-serif
#font.sans-serif     : Arial, Helvetica, sans-serif

将这两行中的注释符#去掉，并将font.sans-serif行的字体名称更改为所需的字体，比如SimSun。

font.family         : sans-serif
font.sans-serif     : SimSun

保存matplotlibrc文件，并重新运行代码，即可看到字体设置已经改变。

注意：如果所需的字体不在系统中，需要先安装相应的字体文件。