我是一名生物信息学家,最近开始学习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之外还有其他任何可以帮助我想象这种图形的信息,请告诉我。?
答案 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
现在我们需要连接节点。您在问题中没有真正说明如何使用边缘生成列表。你发布的内容的可视化看起来就像你描述的那样:两个断开连接的图形。
然而,当我创建一个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)
我得到一张连线图:
因此,我不确定生成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)
生成
确实有6个节点但12个而不是请求的10个边缘。