如何在条件下从数据框中选择多个列和行？

Question

我想选择条件下的行和列，例如：

0是相机，1是视频。 当列== 1时，返回视频数据。 否则返回照片的数据

目的是获取基于视频和照片的单独数据。

代码如下所示。我猜问题出在loc。[i，：]，因为当我将i更改为0时，它成功获取了第一行。但是不知道为什么我不工作。

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)

我希望输出将是视频的5列（“ num_reactions”，“ num_comments”，“ num_shares”，“ num_likes”和“ num_loves”）中的数据。

谢谢。

Answer 1

子集数据集。

示例：

Df_Camera = Dataset[(Dataset['status_type_num'] == 0)]
Df_Video = Dataset[(Dataset['status_type_num'] == 1)]