我有一个数据集，包含4个特征和2个标签，分别是1和0。是否有办法检查这个数据集的所有1400个标签值，并返回它们所属的行？第0到3列是特征，第4列是标签值。

import pandas as pdimport numpy as npimport matplotlib.pyplot as pltfrom sklearn.model_selection import train_test_splitdata = pd.read_csv('data_banknote_authentication.txt', header=None)print(data.head())print(data.tail())print(data.info)X = data.loc[:, :3]Y = data[4]

回答：

您可以使用以下方法查看所有标签值为0的行：

data[data['4'] == 0]

以及标签值为1的行：

data[data['4'] == 1]