python之在 Python 中循环一个函数

如何为下面的数据集循环一个函数。

我只能通过

df =sorteddf[sorteddf.user ==1] 
function(df) 

数据集包含多个用户

time_slot              User     Location 
2017-10-27 15:00:00     1       766 
2017-10-27 15:45:00     1       902 
2017-10-27 16:00:00     1       902 
2017-10-27 10:45:00     2       902 
2017-10-27 12:45:00     2       902  
2017-10-27 16:30:00     2       902  

如何为数据框中的所有用户使用该函数并打印所有用户的输出。

time_slot              User    Location 
2017-10-27 00:00:00     1.0     766.0    
2017-10-27 00:15:00     1.0     961.0    
2017-10-27 00:30:00     1.0     56.0     
2017-10-27 00:45:00     1.0     542.0    
2017-10-27 01:00:00     1.0     324.0    
2017-10-27 01:15:00     1.0     526.0    
2017-10-27 01:30:00     1.0     258.0    
2017-10-27 01:45:00     1.0     532.0    

请您参考如下方法:

您可以为每个用户创建一个循环，如下所示:

users = sorteddf['User'].unique() 
for user in users: 
    print('Output for user {}'.format(user)) 
    function(sorteddf[sorteddf.User == user]) 

这是您要找的东西吗？