Sponsored Content
Top Forums Shell Programming and Scripting 3 column .csv --> correlation matrix; awk, perl? Post 302669049 by R3353 on Tuesday 10th of July 2012 10:59:57 AM
Old 07-10-2012
3 column .csv --> correlation matrix; awk, perl?

Greetings, salutations.

I have a 3 column csv file with ~13 million rows and I would like to generate a correlation matrix. Interestingly, you all previously provided a solution to the inverse of this problem. Thread title: "awk? adjacency matrix to adjacency list / correlation matrix to list" (I haven't posted 5 times yet - can't insert URL)

Brief description of data: columns 1 and 2 contain variables, column 3 contains their correlation. For simplicity and discussion, let's use this example:

Code:
a,a,1
a,b,0.8
a,c,0.5
b,a,0.8
b,b,1
b,c,0.2
c,a,0.5
c,b,0.2
c,c,1

(sidebar - forum users "birei" and "shamrock" provided excellent solutions to remove the a,b = b,a redundancy. Thread title: "awk with existence argument?")

desired output:

Code:
,a,b,c
a,1,0.8,0.5
b,0.8,1,0.2
c,0.5,0.2,1

I currently use the following python script, where the creation of the dictionary and subsequent line-by-line look up is painfully slow for large files.

Code:
import os, time

def matrix(f):
    
    if 'csv' in f and 'matrix' not in f:
        
        i = 0
        on = time.time()
        fp = open(os.path.join(os.getcwd(),f),'r')
        
        data = [x.split(',') for x in fp]
        data.pop(0)
        
        correlations = {}
        for x in data:
            correlations[x[0]] = {}
            variables = correlations.keys()
            variables.sort()
        
        for x in variables:
            for y in variables:
                if x == y:
                    correlations[x][y] = 1
                    i+=1
                else:
                    for row in data:
                        if x in row and y in row:
                            i+=1
                            correlations[x][y] = float(row[2])
        
        data = ''
        for y in variables:
            data += ',%s' %y
        data += '\n'
        
        for x in variables:
            data += '%s,' %x
            for y in variables:
                i+=1
                try:
                    data += '%s,' %correlations[x][y]
                except:
                    data += 'CORRELATION NOT FOUND!,'
            data += '\n'
        fw = open(os.path.join(os.getcwd(),f[:-4]+'_matrix.csv'),'w')
        fw.write(data)
        fw.close()
        
        s = (on - time.time()) / float(1000)
        print '%s complete: %i iterations, %s duration.' %(f,i,s)


for f in os.listdir(os.getcwd()):
    matrix(f)

I am not partial to any particular scripting language. I am, however, partial to speed Smilie. In my experience, awk and perl are the fastest at manipulating these .csv files, but I'm not savvy enough with either. Any suggestions would be appreciated.

Thanks in advance.

Moderator's Comments:
Mod Comment edit by bakunin: I edited the URL in for you. Please use CODE-tags not only for code but also for file content, terminal output, etc.. Thanks.


Will do, thanks bakunin.

Last edited by R3353; 07-10-2012 at 01:54 PM..
 

10 More Discussions You Might Find Interesting

1. Shell Programming and Scripting

correlation coefficient - Awk

Hi guys I have an input file with multiple columns and and rows. Is it possible to calculate correlation of certain value of certain No (For example x of S1 = 112) with all other values (for example start with x 112 corr a 3 of S1 = x-a 0.2 ) INPUT ******* No S1 S2 S3 S4 Sn a 3 ... (2 Replies)
Discussion started by: quincyjones
2 Replies

2. UNIX for Dummies Questions & Answers

How to extract one column from csv file in perl?

Hi everyone, i am new to perl programming, i have a problem in extracting single column from csv file. the column is the 20th column, please help me.. at present i use this code #!C:/perl/bin use warnings; use strict; my $file1 = $ARGV; open FILE1, "<$file1" or die "Can't... (13 Replies)
Discussion started by: kvth
13 Replies

3. Shell Programming and Scripting

two-column data to matrix in AWK

Howdy, I need to convert an association data matrix, currently in a two-column format, into a matrix with numbers indicating the number of associations. I've been looking around for AWK code in the list, but could not find anything. Here's an example of what I want to perform: original... (10 Replies)
Discussion started by: sramirez
10 Replies

4. Shell Programming and Scripting

awk? adjacency matrix to adjacency list / correlation matrix to list

Hi everyone I am very new at awk but think that that might be the best strategy for this. I have a matrix very similar to a correlation matrix and in practical terms I need to convert it into a list containing the values from the matrix (one value per line) with the first field of the line (row... (5 Replies)
Discussion started by: stonemonkey
5 Replies

5. Shell Programming and Scripting

awk: Transpose csv row to column.

Hello, am I new to awk, and I am tryint to: INPUT FILE: "73423555","73423556","73423557","73423558","73423559" OUTPUT FILE: 73423555 73423556 73423557 73423558 73423559 My useless code so far: #!/bin/awk -F ',' BEGIN { i=0; } (8 Replies)
Discussion started by: drbiloukos
8 Replies

6. Shell Programming and Scripting

awk read column csv and search in other csv

hi, someone to know how can i read a specific column of csv file and search the value in other csv columns if exist the value in the second csv copy entire row with all field in a new csv file. i suppose that its possible using awk but i m not expertise thanks in advance (8 Replies)
Discussion started by: giankan
8 Replies

7. Shell Programming and Scripting

Perl code to grep a particular column in CSV format

Hi I want to grep a column 6 & column 7 from a CSV Format file & then i have to find the difference between these columns as these both columns contains date & time in 7/7/2012 9:20 this format . So kindly help me out ASAP. But please kindly dis xls has to be converted in csv format as may... (5 Replies)
Discussion started by: Prateek@123
5 Replies

8. Shell Programming and Scripting

Perl- creating a matrix from a 3 column file

Dear all, I'm new in perl scripting and I'm trying to creating a matrix from a 3 column file sorting data in a particular manner. In the final matrix I need to have the first column "IDs" on the header of the columns and the second column values on the header of each row. And the value fo the... (2 Replies)
Discussion started by: gabrysfe
2 Replies

9. Shell Programming and Scripting

Transpose matrix based on second column using awk

Hi, Is it possible to transpose the matrix like this using awk ? Many thanks in advance Input abc Name_1 0 abc Name_2 1 abc Name_3 2 abc Name_4 0.4 def Name_1 0 def Name_2 9 def Name_3 78 def Name_4 1 Output abc def Name_1 0 ... (4 Replies)
Discussion started by: quincyjones
4 Replies

10. Shell Programming and Scripting

Need awk or Shell script to compare Column-1 of two different CSV files and print if column-1 matche

Example: I have files in below format file 1: zxc,133,joe@example.com cst,222,xyz@example1.com File 2 Contains: hxd hcd jws zxc cst File 1 has 50000 lines and file 2 has around 30000 lines : Expected Output has to be : hxd hcd jws (5 Replies)
Discussion started by: TestPractice
5 Replies
All times are GMT -4. The time now is 11:09 PM.
Unix & Linux Forums Content Copyright 1993-2022. All Rights Reserved.
Privacy Policy