Преобразовать массив ndArray в матрицу - PullRequest
0 голосов
/ 27 сентября 2018

У меня загружен набор данных из файла -

np.array([(5.1, 3.5, 1.4, 0.2, 'Iris-setosa'),
          (4.9, 3., 1.4, 0.2, 'Iris-setosa'),
          (4.7, 3.2, 1.3, 0.2, 'Iris-setosa'),
          (4.6, 3.1, 1.5, 0.2, 'Iris-setosa'),
          (5., 3.6, 1.4, 0.2, 'Iris-setosa')])

Как получить матрицу из первых 4 столбцов -

np.array([[5.1, 3.5, 1.4, 0.2],
          [4.9, 3., 1.4, 0.2],
          [4.7, 3.2, 1.3, 0.2],
          [4.6, 3.1, 1.5, 0.2],
          [5., 3.6, 1.4, 0.2]])

и метки в другом массиве -

np.array(['Iris-setosa', 'Iris-setosa', 'Iris-setosa', 'Iris-setosa',
          'Iris-setosa'])

1 Ответ

0 голосов
/ 27 сентября 2018
import pandas as pd

data=np.array([(5.1, 3.5, 1.4, 0.2, 'Iris-setosa') ,
 (4.9, 3. , 1.4, 0.2, 'Iris-setosa'),
 (4.7, 3.2, 1.3, 0.2, 'Iris-setosa'),
 (4.6, 3.1, 1.5, 0.2, 'Iris-setosa'),
 (5. , 3.6, 1.4, 0.2, 'Iris-setosa')])

df= pd.DataFrame(data)
first4col = df.iloc[:,0:3]
tags=df.iloc[:,4]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...