Доступ к столбцу в наборе данных Scikit

Я загружаю встроенный набор данных Бостона в библиотеке Scikit как:

from sklearn.datasets import load_boston bdata = load_boston()

Я хочу извлечь все значения в первом столбце, который называется CRIM. Я написал строку вроде: plt.scatter(bdata.CRIM,bdata.target,color='blue')

Но я получаю сообщение об ошибке, так как "AttributeError: объект" Bunch "не имеет атрибута CRIM"

Как я могу получить доступ к элементам столбца под названием "CRIM"?

1 ответ

Решение

Значения данных хранятся в атрибуте данных, чтобы получить доступ ко всем значениям первого столбца, вы используете синтаксис numpy's slice. В этом случае вы хотите:

plt.scatter(bdata.data[:,0],bdata.target,color='blue')

Двоеточие указывает, что вы выбираете "все" для конкретной оси (в данном случае это первая ось, в данном случае "строки"), а 0 указывает, что вы хотите первый элемент для всех строк - это первый столбец.

Другие вопросы по тегам