Значение df.head (5) состоит в том, что он показывает первые 5 строк кадра данных. Он не будет добавлять любое количество строк в цикле. Вам нужно использовать переменную
Я думаю, вы имеете в виду, что эта программа работает следующим образом:
import pandas as pd
df = pd.read_csv("train.csv")
i = 5
#df.shape[0] gives the number of rows
while(i< df.shape[0]):
check = input(' Do You Wish to continue yes/no: ')
if check == 'yes' :
print(df.head(i))
i+=5#increment 5
else :
#if input is not 'yes' end loop
break;