要求用户继续查看接下来的5行数据

时间:2018-06-14 22:35:29

标签: python pandas

我试图获取接下来的5个数据行,而不是每次都写一个循环([:5],[5:10],[10:15]等等)。我有一些想法,但我一直坚持这个,有什么方法可以让用户显示下一个5?

def raw_data(df):
ask_user = input('Would you like to view more raw data for the city selected? \nPrint yes or no: ')
while True:
    if ask_user == 'yes':
        return df.iloc[:5]
    else:
        break

1 个答案:

答案 0 :(得分:3)

添加每次运行函数时累积的变量。 (让我们称之为runs)在函数外部初始化它,并在函数内部添加一个函数。从这里开始,乘以5得到你的范围。

runs = 0
def raw_data(df):
    while True: 
        ask_user = input('Would you like to view more raw data for the city selected? \nPrint yes or no: ') 
        if ask_user == 'yes': 
            runs += 1 #Adds 1 to current value, same as runs = runs + 1
            return df.iloc[(x-1)*5:x*5]
        elif ask_user == 'no':
            return