#test 2 prep part 1

def studentID(major, ID):
	return major[0:4].upper()+ID


def newID_list(major_list, ID_list):
	new_list=[]
	size=len(major_list)
	for i in range(size):
		new_id=studentID(major_list[i],ID_list[i])
		new_list.append(new_id)
	return new_list
def process(new_id_list, major):
	major_list=[]
	not_major_list=[]
	for item in new_id_list:
		if(item[0:4]==major[0:4].upper()):
			major_list.append(item)
		else:
			not_major_list.append(item)
	return major_list,not_major_list

def make_list_string(size):
	#this function creates a list of strings from user input

	my_list=[]
	for i in range(size):
		new_str = input("enter string ")
		my_list.append(new_str)

	return my_list

def main():
	size=int(input("size "))
	print("enter majors")
	majors=make_list_string(size)
	print("enter ID's")
	ID=make_list_string(size)

	print("majors", majors)
	print("IDs",ID)
	new_id=newID_list(majors,ID)
	print("new ID's",new_id)
	your_major=input("enter your major ")
	m_list, nm_list=process(new_id, your_major)
	print("students with your major", m_list)
	print("nor your major", nm_list)
main()