csv_writer.py

import csv


def write_csv(filename, header, data):
	"""Write the provided data to the CSV file.
	:param str filename: The name of the file to which the data should be written.
	:param list header: The header for the columns in the csv file.
	:param list data: The list of list mapping the values to the columns
	"""

	try:
		with open(filename, 'w', newline='') as csv_file:
			csv_writer = csv.DictWriter(csv_file, fieldnames=header)
			csv_writer.writeheader()
			csv_writer.writerows(data)

	except (IOError, OSError) as csv_file_error:
		print(f"Unable to write the contents to csv file. Exception: {csv_file_error}")

if __name__ == '__main__':
	header = ['name', 'age', 'gender']
	data = [{'name': 'Richard', 'age': 32, 'gender': 'M'},
			{'name': 'Numzil', 'age': 21, 'gender': 'F'},
			{'name': 'Melinda', 'age': 25, 'gender': 'F'}]
	filename = 'sample_output.csv'
	write_csv(filename, header, data)