Matrix of sequence in Python

Question

I have this code:

import random
def create_matrix(row, col):
    matrix = [["".join([random.choice("ATCG") for j in range(random.randint(1, 10))])
                  for _ in range(col)]
                  for _ in range(row)]
    return matrix
print(create_matrix(5, 4))

And I get this output:

[['CGCT', 'CATGC', 'GCAACTATA', 'AA'], ['TTTGGGAGTA', 'GGGCTGAA', 'GTT', 'GCTCGC'], ['ATTGT', 'CCG', 'ACC', 'GTG']

How can I get the same output but arranged in a matrix 3x4? Something like this but with the previous output.

1   2   3   4
5   6   7   8
9   10  11  12
13  14  15  16
17  18  19  20

Here there are some useful trick: delftstack.com/howto/python/print-matrix-python — Glauco
– Glauco, Commented Oct 7, 2021 at 15:13

Ewan Brown · Accepted Answer · 2021-10-07 15:06:47Z

1

If you're not concerned about spacing things nicely, a simple for loop to print each row at a time would work. (Try Online)

import random
def create_matrix(row, col):
    matrix = [["".join([random.choice("ATCG") for j in range(random.randint(1, 10))])
                  for _ in range(col)]
                  for _ in range(row)]
    return matrix

matrix = create_matrix(5, 4)

for r in matrix:
    print(r)

With output:

['TTTG', 'TCT', 'ACTCGAGCTA', 'AA']
['TTAC', 'GT', 'T', 'CGGAC']
['GGCATGT', 'CCCTCCGTCC', 'GTCGAA', 'GTCATGTAG']
['TC', 'TTACGGTCT', 'TGGCAAAA', 'AGCCGGGGA']
['A', 'C', 'CT', 'AA']

answered Oct 7, 2021 at 15:06

Ewan Brown

6373 silver badges13 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

David Arteaga M. Over a year ago

Great! It was easy. Thank you!

Ftagliacarne · Accepted Answer · 2021-10-07 15:08:52Z

0

If you use numpy, the output will be better formatted. But basically what you have is correct, it's just a matter of the code is outputted.

import numpy as np
print(np.array(create_matrix(5, 4)))

Will produce the following output:

[['CT' 'CA' 'CCAGTG']
 ['ACAGAA' 'AAT' 'C']
 ['ATCCAGAA' 'ATCGACCGTG' 'T']
 ['GACA' 'AACCGTAG' 'GCCACAGAC']
 ['TCAATT' 'TTTCAG' 'TACGCCTGA']]

answered Oct 7, 2021 at 15:08

Ftagliacarne

79210 silver badges22 bronze badges

Collectives™ on Stack Overflow

Matrix of sequence in Python

2 Answers 2

1 Comment

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related