Skip to content
Snippets Groups Projects
cluster_modules.py 1.02 KiB
Newer Older
#!/usr/bin/python
import csv

def get_data(filename):
    '''function to read the data form the input csv file to use in the analysis'''
    reader = [] # Just in case the file open fails
    with open(filename, 'rb') as f:
        reader = csv.reader(f,delimiter=',')                      
        #returns all the data from the csv file in list form
        #f.close() # May need to close the file when done
        return list(reader)  # only return the reader when you have finished.

your_list = []
your_list += get_data('modules-anselm.csv')
your_list += get_data('modules-salomon.csv')
#print your_list
your_list.sort()

a=[["python/2.8.1",1],["python/2.9.1",2],["python/2.8.1",4],["python/3.0.1",4]]
counts = dict()
for i in your_list:
  #print i[0]
  #print int(i[1])
  counts[i[0]]=counts.get(i[0], 0) + int(i[1])

#print counts

c=[
"---",
"--A",
"-S-",
"-SA",
"U--",
"U-A",
"US-",
"USA",
]

print "| Module | Clusters |"
print "| ------ | -------- |"

for m,i in counts.iteritems():
  print "| %s | <code>%s</code> |" % (m,c[i])