Question

如何在Python中创建具有功能和标签的新矩阵？例如。在scikit-learn中：如何“看到”（打印）具有功能+标签的新表：[5。 2.5 1.5 0.2 0]，最后一个0是此样本的标签。另外：我创建了训练和测试数据集-如何通过训练或测试集成员“标记”基本数据？（我是Python新手）

feature matrix
    [[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]]

    label vector
    [0 1 0 2]

     how can I "put" together into one matrix?
    [[5.1 3.5 1.4 0.2 0]
     [4.9 3.  1.4 0.2 1]
     [4.7 3.2 1.3 0.2 0]
     [4.6 3.1 1.5 0.2 2]]

代码：

from sklearn.datasets import load_iris
iris_dataset = load_iris()

print ((iris_dataset['data']))
print ((iris_dataset['target']))

print("Type of data: {}".format(type(iris_dataset['data'])))
print("Type of target: {}".format(type(iris_dataset['target'])))


Type of data: <class 'numpy.ndarray'>
Type of target: <class 'numpy.ndarray'>

Answer 1

您可以使用numpy hstack以这种方式轻松地连接数组。

from sklearn.datasets import load_iris
iris_dataset = load_iris()

print(iris_dataset['data'].shape)
print(iris_dataset['target'].shape)

import numpy as np
result = np.hstack((iris_dataset['data'], iris_dataset['target'].reshape(-1, 1)))

print(result.shape)

#Output:
(150, 4)
(150,)
(150, 5)

创建具有矩阵+向量（特征+标签）的矩阵

1 个答案: