allpy: a3df15a42f9d test/freqs.py

allpy

blocks3d(): fix a bug when finding pair cores, all alternative cores should be considered!!

author	boris (kodomo) <bnagaev@gmail.com>
date	Sat, 26 Feb 2011 18:06:45 +0300
parents
children

line source

1 """Read alignment on stdin. Print CSV table of letter frequences on stdout.

3 from allpy import protein

4 from allpy.data import codes

5 import sys

7 sys.stderr.write(__doc__)

9 def freq(monomer):

10 amount = freqs.get(monomer)

11 if amount:

12 return 100.0 * amount / width

13 return ""

15 aln = protein.Alignment().append_file(sys.stdin)

16 monomers = [code1 for code1, modified, _, _ in codes.protein if not modified]

17 monomers += ["-"]

18 width = len(aln.sequences)

19 print ", ".join(map(str, monomers))

20 for column in aln.columns_as_lists():

21 freqs = {}

22 for monomer in column:

23 if monomer:

24 monomer = monomer.code1

25 else:

26 monomer = "-"

27 freqs[monomer] = freqs.get(monomer, 0) + 1

28 print ", ".join(map(str, map(freq, monomers)))