Работа с jupiter notebook

import pandas as pd
import numpy as np
exams = ['python1', 'python2', 'python3', 'python4', 'python5']
df = pd.DataFrame(columns = ['ФИО', '№_группы'] + exams)
 
while True:
    student = [input('Введите фамилию и инициалы студента:\n')]
    student.append(input('Введите номер группы:\n'))
    student.extend(list(map(int,input('Введите оценки за экзамен через пробел:\n').split())))
 
    df.loc[len(df)] = student
    query = input('Еще есть студенты? -> 1(да) / 0(нет): ')
    if query == '0':
        break
print()
df[exams] = df[exams].astype(np.float16)
print(df)
print()
df_4 = df[df[exams].mean(axis=1) > 4][['ФИО', '№_группы']]
if df_4.empty:
    print('Таких студентов нет')
else:
    print(df_4)

import pandas as pd
import numpy as np
exams = ['exam_1']
df = pd.DataFrame(columns = ['ФИО', '№_группы'])
cnt_col = df.shape[1]
while True:
    student = [input('Введите фамилию и инициалы студента:\n')]
    student.append(input('Введите номер группы:\n'))
    student.extend(list(map(int,input('Введите оценки за экзамен через пробел:\n').split())))
    k = len(student)
    if k > cnt_col:
        exams = []
        while k > cnt_col:
            exams.append('exam_' + str(cnt_col-1))
            cnt_col += 1
    df = pd.concat([df,pd.DataFrame(columns=exams)])  
    df.loc[len(df)] = student
    query = input('Еще есть студенты? -> 1(да) / 0(нет): ')
    if query == '0':
        break
print()
df.fillna(0, inplace=True)
df[exams] = df[exams].astype(np.int8)
print(df)
print()
df_4 = df[df[exams].mean(axis=1) > 4][['ФИО', '№_группы']]
if df_4.empty:
    print('Таких студентов нет')
else:
    print(df_4)

import pandas as pd
import numpy as np
exams = ['exam_1']
df = pd.DataFrame(columns = ['ФИО', '№_группы'])
cnt_col = df.shape[1]
while True:
    student = [input('Введите фамилию и инициалы студента:\n')]
    student.append(input('Введите номер группы:\n'))
    student.extend(list(map(int,input('Введите оценки за экзамены через пробел:\n').split())))
    k = len(student)
    if k > cnt_col:
        exams = []
        while k > cnt_col:
            exams.append('exam_' + str(cnt_col-1))
            cnt_col += 1
        df = pd.concat([df,pd.DataFrame(columns=exams)])  
    df.loc[len(df)] = student + [np.nan] * (cnt_col-k)
    query = input('Еще есть студенты? -> 1(да) / 0(нет): ')
    if query == '0':
        break
print()
# df.fillna(0, inplace=True)
df.iloc[:, 2:] = df.iloc[:, 2:].astype(np.float16)
print(df)
print()
df_4 = df[df.iloc[:, 2:].mean(axis=1) > 4][['ФИО', '№_группы']]
if df_4.empty:
    print('Таких студентов нет')
else:
    print(df_4)