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

Question

我有一个包含 4 个特征和 2 个标签（1 和 0）的数据集。有没有办法可以检查该数据集的所有 1400 个标签值并返回它属于哪一行？第 0-3 列是特征，第 4 列是标签值。

import pandas as pd

import numpy as np

import matplotlib.pyplot as plt

from sklearn.model_selection import train_test_split

data = 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]