I would like to choose the rows and columns under condition, for example:
0 is camera, 1 is video. when the column == 1, return the data of video. else return the data of photo
The purpose is to get separate data based on video and photo.
The code shows below. I guess the problem is from loc.[i, :] because when I changed i to 0, it grab the first row successfully. But don't know why i doesn't work.
for i in range(len(dataset)): if dataset['status_type_num'][i] == 1: video_data = dataset[['num_reactions', 'num_comments', 'num_shares', 'num_likes', 'num_loves']].loc[i, :] print(video_data)
I expect output would be the data from 5 columns('num_reactions', 'num_comments', 'num_shares', 'num_likes', 'num_loves') of video.