python - 读入 np 数组不起作用

Question

希望一切顺利...我正在将数据集输入到 sklearn 算法中进行分类，但找不到任何简单的数据集来开始，所以我自己制作了数据集。但有一个问题...

import numpy as np
import random

type_1 = [random.randrange(0, 30, 1) for i in range(50)]
type_1_label = [1 for i in range(50)]

type_2 = [random.randrange(31, 75, 1) for i in range(50)]
type_2_label = [-1 for i in range(50)]

zipped_1 = zip(type_1, type_1_label)
zipped_2 = zip(type_2, type_2_label)

ready = np.array(zipped_1)
print(ready[1])

这里的问题是，当我用类型一压缩输入一个标签时，输出是一个数组，由具有两个索引的数组组成，正如预期的那样，然后我需要将其输入到一个返回 IndexError: 的 numpy 数组中数组的许多索引对我来说没有意义； numpy 肯定可以读取 2x2 数组的 N 维数组函数吗？任何帮助将不胜感激!

Answer 1

您可以直接创建您想要的 NumPy 数组作为结果:

ready1 = np.random.randint(0, 30, size=(50, 2))
ready1[:, 1] = 1

ready2 = np.random.randint(31, 71, size=(50, 2))
ready2[:, 1] = -1