绘制序列图及其反向补码

时间:2015-10-13 11:09:07

标签: python graph graphviz bioinformatics

我是一名生物信息学家,最近开始学习python,我对绘制Graph有兴趣。我有一组节点和边缘。

节点

set(['ACG', 'ATC', 'GAT', 'CGG', 'CGT', 'AAT', 'ATT', 'GGA', 'TTC', 'CCG', 'TCC', 'GAA'])

边缘

[('ACG', 'CGG'), ('CGG', 'GGA'), ('GGA', 'GAA'), ('GAA', 'AAT'), ('AAT', 'ATC'), ('GAT', 'ATT'), ('ATT', 'TTC'), ('TTC', 'TCC'), ('TCC', 'CCG'), ('CCG', 'CGT')]

当我使用上述信息构建法线图时,我得到12个节点和10个边,即使用以下函数的两个断开图。

def visualize_de_bruijn():
    """ Visualize a directed multigraph using graphviz """
    nodes = set(['ACG', 'ATC', 'GAT', 'CGG', 'CGT', 'AAT', 'ATT', 'GGA',      'TTC', 'CCG', 'TCC', 'GAA'])
    edges = [('ACG', 'CGG'), ('CGG', 'GGA'), ('GGA', 'GAA'), ('GAA', 'AAT'), ('AAT', 'ATC'), ('GAT', 'ATT'), ('ATT', 'TTC'), ('TTC', 'TCC'), ('TCC', 'CCG'), ('CCG', 'CGT')]
    dot_str = 'digraph "DeBruijn graph" {\n'
    for node in nodes:
        dot_str += '  %s [label="%s"] ;\n' % (node, node)
    for src, dst in edges:
        dot_str += '  %s -> %s ;\n' % (src, dst)
    return dot_str + '}\n'

在生物学中,我们有这种互补碱基配对的概念,其中A = T,T = A,G = C和C = G.所以免费提供ACG'是' TGC'反向补充' ACG' =' CGT'即我们颠倒补语。

在节点列表中,我们看到12个节点,其中6个节点是彼此反向互补的,即

ReverseComplement('ACG') = CGT 
ReverseComplement('CGG') = CCG 
ReverseComplement('GGA') = TCC   
ReverseComplement('GAA') = TTC 
ReverseComplement('AAT') = ATT 
ReverseComplement('ATC') = GAT

现在我想构建一个图,其中有六个节点,一个节点应该有自己的值和它的反向补码值,总共10个边,即图不应该断开。如何在python中使用graphviz可视化此图形。如果除了graphviz之外还有其他任何可以帮助我想象这种图形的信息,请告诉我。?

1 个答案:

答案 0 :(得分:2)

我不确定你在这里想要完成什么(所以要注意你可能有XY problem),但是让我们回答你的问题并看看它在哪里得到我们。

  

节点应具有自己的值及其反向补码值

因此我们需要一些对象来存储序列和该序列的反向补码。

different ways of making the reverse complement of a sequence。作为生物信息学家,您应该真正使用适合生物信息学的库,即BioPython

然后制作反向补充看起来像这样:

from Bio.Seq import Seq

seq = 'ATCG'
print str(Seq(seq).reverse_complement()) # CGAT

但是生成Seq对象可能对此问题有点过分,所以我只使用下面的标准字典。我们还希望将Node个对象相互比较,因此我们需要覆盖__eq__。因为我们想要制作set个唯一Node个对象,所以我们也需要实现__hash__。对于漂亮的打印,我们还实现了__str__

class Node(object):
    def __init__(self, seq):
        self.seq = seq
        self.revcompl = self.revcompl()

    def revcompl(self):
        complement = {'A': 'T', 'C': 'G', 'G': 'C', 'T': 'A'}
        return "".join(complement.get(base, base) for base in reversed(self.seq))

    def __eq__(self, other):
        return self.seq == other.seq or self.revcompl == other.seq

    def __hash__(self):
        return hash(self.seq) ^ hash(self.revcompl)

    def __str__(self):
        return '({}, {})'.format(self.seq, self.revcompl)

现在我们可以将我们的set或original节点转换为我们的新节点列表及其反向补码。

nodes = set(['ACG', 'ATC', 'GAT', 'CGG', 'CGT', 'AAT', 'ATT', 'GGA', 'TTC', 'CCG', 'TCC', 'GAA'])
newnodes = set(Node(seq) for seq in nodes)
assert len(newnodes) == 6

现在我们需要连接节点。您在问题中没有真正说明如何使用边缘生成列表。你发布的内容的可视化看起来就像你描述的那样:两个断开连接的图形。

two disconnected graphs

然而,当我创建一个DeBruijn图时,我会成对地比较所有序列,看看它们之间是否有任何重叠,创建一个邻接列表,并从中生成graphviz的DOT代码。

from itertools import product

def suffix(seq, overlap):
    return seq[-overlap:]

def prefix(seq, overlap):
    return seq[:overlap]

def has_overlap_seq(seq1, seq2, overlap=2):
    if seq1 == seq2:
        return False
    return suffix(seq1, overlap) == prefix(seq2, overlap)

def get_adjacency_list_seqs(sequences, overlap=2):
    for seq1, seq2 in product(sequences, repeat=2):
        if has_overlap_seq(seq1, seq2, overlap):
            yield seq1, seq2

def make_dot_plot(adjacency_list):
    """Creates a DOT file for a directed graph."""
    template = """digraph "DeBruijn graph"{{
{}
}}""".format
    edges = '\n'.join('"{}" -> "{}"'.format(*edge) for edge in adjacency_list)
    return template(edges)

如果我为您的原始nodes

执行此操作
seq_adjacency_list = get_adjacency_list_seqs(nodes)
print make_dot_plot(seq_adjacency_list)

我得到一张连线图:

single connected graph

因此,我不确定生成edges列表的原始实现中是否存在错误,或者您是否尝试完全执行其他操作。

现在继续前进,我们可以调整前面的序列字符串代码,也可以使用我们之前创建的Node个对象。

def has_overlap_node(node1, node2, overlap=2):
    if node1 == node2:
        return False
    return suffix(node1.seq, overlap) == prefix(node2.seq, overlap) or \
           suffix(node1.seq, overlap) == prefix(node2.revcompl, overlap) or \
           suffix(node1.revcompl, overlap) == prefix(node2.seq, overlap) or \
           suffix(node1.revcompl, overlap) == prefix(node2.revcompl, overlap)

def get_adjacency_list_nodes(nodes, overlap=2):
    for node1, node2 in product(nodes, repeat=2):
        if has_overlap_node(node1, node2, overlap):
            yield node1, node2

应用此

nodes_adjacency_list = get_adjacency_list_nodes(newnodes)
print make_dot_plot(nodes_adjacency_list)

生成

nodes plot

确实有6个节点但12个而不是请求的10个边缘。

相关问题