Search code examples
pythoncsvadjacency-matrixcytoscape

Convert adjacency matrix to a csv file


I want to convert the adjacency matrix output from ARACNE into a csv file using python (or possibly R).

The adj file is set up to show one gene one the right and each of its interactions with other genes. For example:

A B 0.4 C 0.3
B C 0.1 E 0.4
C D 0.2 E 0.3

So above, A and B interact with each other and the value of that interaction is 0.4. A and C interact with each other and the value is 0.3 and so on.

I want to change the layout so I get...

A B 0.4
A C 0.3
B C 0.1
B E 0.4
C D 0.2
C E 0.3

Basically I want a list of all interacting nodes and the corresponding values so that I can upload the file to Cytoscape and plot a network.


Solution

  • A simple way to do this using csv module -

    import csv
    with open('<inputfile>','r') as f , open('<outputfile>','w') as of:
        reader = csv.reader(f, delimiter=' ')
        writer = csv.writer(of, delimiter=' ')
        for lines in reader:
                key = lines.pop(0)
                for i in range(0,len(lines),2):
                        writer.writerow([key, lines[i], lines[i+1]])
    

    Example/Demo -

    My a.csv -

    A B 0.4 C 0.3
    B C 0.1 E 0.4
    C D 0.2 E 0.3
    

    Code -

    >>> import csv
    >>> with open('a.csv','r') as f , open('b.csv','w') as of:
    ...     reader = csv.reader(f, delimiter=' ')
    ...     writer = csv.writer(of, delimiter=' ')
    ...     for lines in reader:
    ...             key = lines.pop(0)
    ...             for i in range(0,len(lines),2):
    ...                     writer.writerow([key, lines[i], lines[i+1]])
    

    Output in b.csv -

    A B 0.4
    A C 0.3
    B C 0.1
    B E 0.4
    C D 0.2
    C E 0.3