31 lines
608 B
Python
31 lines
608 B
Python
import numpy as np
|
|
import csv
|
|
|
|
# 定义回归方程参数
|
|
w = 1.35
|
|
b = 2.89
|
|
|
|
# 生成x值范围
|
|
x_min = 0
|
|
x_max = 10
|
|
|
|
# 生成100个在x轴附近的点
|
|
x = np.linspace(x_min, x_max, 100)
|
|
|
|
# 根据回归方程计算y值
|
|
y = w * x + b
|
|
|
|
# 添加一些噪声，使数据更真实
|
|
y += np.random.normal(scale=0.5, size=y.shape)
|
|
|
|
# 将x和y合并成一个二维数组
|
|
data = np.column_stack((x, y))
|
|
|
|
# 将数据保存到CSV文件
|
|
with open('data1.csv', 'w', newline='') as csvfile:
|
|
writer = csv.writer(csvfile)
|
|
# 写入表头
|
|
# writer.writerow(['x', 'y'])
|
|
# 写入数据
|
|
writer.writerows(data)
|