Код на вывод среднего балла по каждой группе

[Jorsan]

Здравствуйте, помогите пожалуйста , у меня есть код, который берёт текст из текстового файла выводит средний балл каждого ученика пиша его группу в другой текстовый файл.
Но мне нужно, чтобы код находил средний балл всех учеников по группе, если таковых несколько.
Test input:

Васильев;П1-20;4;4;5
Герасимов;П1-18;5;5;5
Лаухин;П1-20;3;4;5

Test output:

П1-20: 4.33
П1-18: 5.00
П1-20: 4.00

Собственно сам код:

groups = ["П1-18", "П2-18", "П1-19", "П2-19", "П3-19", "П1-20", "П2-20", "П3-20", "П1-21", "П2-21"]
lst = []
students = list()
with open('students.txt', 'r', encoding='utf-8') as f:
    for line in f:
        lst.append(line.strip().split('\n'))
        parts = line.split(';')
        group = parts[1]
        rating = list(map(int, parts[2:]))
        mean = sum(rating) / len(rating)

        info = list()
        info.append(group)
        info.extend(rating)
        info.append(mean)

        students.append(info)
f = open('sredball.txt', 'w+')
for st in students:
    print('{}: {:.2f}'.format(st[0], st[-1]))

print()

Вывод:

П1-20: 4.33
П1-18: 5.00
П1-20: 4.00

Ожидаемый вывод:

П1-18 5.0
П2-18 0.0
П1-19 0.0
П2-19 0.0
П3-19 0.0
П1-20: 4.2
П2-20 0.0
П3-20 0.0
П1-21 0.0
П2-21 0.0

[VeryStupidPerson]

Вот работающий код:

groups = []
marks = []

with open('students.txt, 'r') as f:
    for line in f:
        data = line.replace('\n', '').split(';')
        if not data[1] in groups:
            groups.append(data[1])
            marks.append([])
        for i in data[2::]:
            marks[groups.index(data[1])].append(int(i))

with open('rezult.txt', 'w') as f:
    for i in range(len(groups)):
       f.write(groups[i] + ' ' + str(round(sum(marks[i]) / len(marks[i]), 1)) + '\n')

Если нужно, могу прокомментировать, что в нём происходит.

П.С. В файл записываются только те группы, которые есть в входном файле, так что вывод состоит только из существующих групп.