numpy中hstack和reshape的使用示例

data = []
label =  []
np.random.seed(0)

for i in range(5):
    x1 = np.random.uniform(-1, 1)
    x2 = np.random.uniform(0, 2)
    if x1**2+x2**2 <= 1:
        data.append([np.random.normal(x1, 0.1), np.random.normal(x2, 0.1)])
        label.append(0)
    else:
        data.append([np.random.normal(x1, 0.1), np.random.normal(x2, 0.1)])
        label.append(1)

print("data:", data)

print("label:", label)

print("np.hstack(data):", np.hstack(data))

data = np.hstack(data).reshape(-1, 2)  #hstack把list(data)推叠成reshape自动推断numpy数组为n*2维
label = np.hstack(label).reshape(-1, 1)
print("data:", data)
print("label:", label)

运行结果：