csv, numpy interop

scipy_io_alternatives.py

"""
Alternatives to working with scipy
"""

import argparse
import csv
import numpy
import scipy.io as sio
import sys


def mat_to_csv(mat_file_name, matrix_var_name, csv_file_name = None):
	"""
	Convert .mat file to .csv

		- mat_file_name : foo.mat or whatever
		- matrix_var_name : whatever you stored your matrix in matlab as
		- csv_file_name : output file name [optional]
	"""
	matrix = sio.loadmat(mat_file_name, {})[matrix_var_name]

	if csv_file_name is None:
		csv_file_name = mat_file_name.replace('.mat', '') + '.csv'

	numpy.savetxt(csv_file_name, matrix, delimiter=',')

def csv_to_np_array(csv_file_name):
	"""
	Load .csv file contents into numpy array object

		- csv_file_name : foo.csv or w/e
	"""
	return numpy.loadtxt(csv_file_name, delimiter = ',')

def _set_up_argparser():
	parser = argparse.ArgumentParser()
	parser.add_argument('--test', dest='test', action='store_true', default=None, help='runs the test')
	return parser


if __name__ == '__main__':
	arg_parser = _set_up_argparser()
	parsed = arg_parser.parse_args()

	if not parsed.test:
		print 'Import and use please'
		sys.exit(1)

	test_arr = numpy.zeros((2, 2))
	sio.savemat('test.mat', {'test' : test_arr})

	mat_to_csv('test.mat', 'test')

	if numpy.array_equal(csv_to_np_array('test.csv'), test_arr):
		print 'Tests passed'
		sys.exit(0)

	print 'FUCK UP'
	sys.exit(1)