Assignment 1(Python)

Assignment1.py

#################################
#         Assignment 1          #
#    Read every mat file in     #  
#    a folder and plot the      #
#    annotation counts they     #
#    contain.	                #
#                               #
#   Author : Nithikan (Bam)     #
#   Date : 02.04.2018           #
#                               #
#################################

import os
import scipy.io as sio
import matplotlib.pyplot as plt

foldername = '.'
filtername = 'mat'
foldercontents = os.listdir(foldername)


filteredcontents = [i for i in foldercontents if i.split('.')[-1] == filtername] 

print(filteredcontents)

annCount = []

for i in filteredcontents:
	print ("D:/Coop/UCF_CC_50/" +str(i))
	data = sio.loadmat("D:/Coop/UCF_CC_50/" +str(i))
	len(data["annPoints"])
	print(len(data["annPoints"]))
	annCount.append(len(data["annPoints"]))

print(annCount)


#plt.plot(annCount)
print(len(annCount))
x = [i for i in range(len(annCount))]
plt.bar(x, annCount)
plt.show()