有没有办法使用 pandas 或 numpy 检查数据集的所有标签值？

Question

我有一个包含 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]

Answer 1

您可以通过以下方式查看所有 0 行：

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

和 1：

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