Загрузка набора данных Iris из файла CSV?

Question

Я практикую обработку данных с помощью Scikit learn и смотрю на вероятность классификации. Я успешно запустил модель, используя набор данных из import dataset, теперь я хочу попробовать сделать то же самое с файлом CSV, поэтому я скачал тот же набор данных и пытаюсь загрузить его в свой код.

iris = np.loadtxt('./iris.csv', delimiter=',', skiprows=1)
X = iris.data[:, 0:2]
y = iris.target 

Однако я получаю сообщение об ошибке ValueError: could not convert string to float: 'setosa' Я понимаю, что это из CSV, так как это имя цветка, есть ли другой способ импортировать этот файл CSV, чтобы эта проблема не вопрос

Answer 1

Для этого вы можете использовать pandas:

data = pandas.read_csv("iris.csv")
data.head() # to see first 5 rows
X = data.drop(["target"], axis = 1)
Y = data["target"]

или вы можете попробовать (я бы лично рекомендовал использовать pandas)

from numpy import genfromtxt
my_data = genfromtxt('my_file.csv', delimiter=',')