Я создал массив с метками, используя приведенный ниже код.Когда я запускаю X[y == 0]
, он выдает мне только те значения в массиве X
, для которых y
равно 0. Однако, если у меня просто было X
и я не знал, что y
было именем столбца, как мне это сделать?получить имя столбца y
и значения для y
, которые равны 0 и 1 от X?
Что-то вроде X.column_names
или X.labels
- это то, что я ищу.
import torch
import numpy as np
from sklearn import datasets
X, y = datasets.make_blobs(n_samples = n_pts, random_state = 123, centers = centers, cluster_std = 0.4)