make-yaml-with-csv.py

import csv

my_path="paintings-for-python-yaml.csv"
newdata=[]
mydata=[]

with open(my_path, 'r') as f:
	reader = csv.reader(f)
	mydata = list(reader)

mydata.remove(mydata[0])
print(len(mydata))

print(mydata[2][1])
# Removes all lines where Group Number is empty
# So all non-image lines; converts Group Number to integer

for i in range(0,len(mydata)):
	if mydata[i][1]!='':
		mydata[i][1]=int(mydata[i][1])
		newdata.append(mydata[i])
#print(newdata[190])

print(len(newdata))
print(newdata[189])
print(type(mydata[1]))
print(type(newdata[1]))

def SetSizeClass(width):
	'''Returns a size class based on the width of the image'''

	width=int(float((width))
	if width == 0:
		size ="missing"
	elif width >= 1 and width < 29:
		size="smallest"
	elif width >= 29 and width < 49:
		size="smaller"
	elif width >= 49 and width < 99:
		size="medium"
	elif width >= 99 and width < 149:
		size="larger"
	elif width >= 149 and width < 199:
		size="pretty big"
	else:
		size ="huge"
	return(size)

#sizeclass=SetSizeClass(newdata[0][5])
#print(sizeclass)

for i in range(0,len(newdata)):
	newdata[i][9]=SetSizeClass(newdata[i][5])
	print(newdata[i])

paintings-for-python-yaml.csv

yaml-as-function-3.py

newdata=['', 1, 'Chain Test', '1981', 'paint on canvas', '54.5', '117', '01-Chain.JPG', '\n']

def SetSizeClass(width):
	'''Returns a size class based on the width of the image
	'''

	width=int(float(width))

	if width == 0:
		size="missing"
	elif width < 29:
		size="smallest"
	elif width >= 29 and width < 49:
		size="smaller"
	elif width >= 49 and width < 99:
		size="medium"
	elif width >= 99 and width < 149:
		size="larger"
	elif width >= 149 and width < 199:
		size="pretty big"
	else:
		size ="huge"
	return(size)

sizeclass=SetSizeClass(newdata[5])
print(sizeclass)