在numpy用array生成游乐园订单数据并保存数据
时间: 2024-02-27 11:57:50 浏览: 24
好的,以下是用numpy生成游乐园订单数据并保存数据的示例代码:
```python
import numpy as np
# 生成 1000 条订单数据,每条订单包含 4 个属性:订单编号、游乐项目、数量和金额
order_data = np.zeros((1000, 4), dtype=np.int32)
order_data[:, 0] = np.arange(1000) + 1 # 订单编号从 1 开始
order_data[:, 1] = np.random.randint(1, 6, size=1000) # 随机生成游乐项目编号
order_data[:, 2] = np.random.randint(1, 6, size=1000) # 随机生成购买数量
order_data[:, 3] = order_data[:, 1] * order_data[:, 2] * 10 # 计算订单金额
# 保存数据到文件
np.savetxt('order_data.csv', order_data, delimiter=',', fmt='%d')
```
这段代码首先使用 `numpy.zeros()` 函数创建一个大小为 (1000, 4) 的二维数组,表示有 1000 条订单数据,每条订单包含 4 个属性。然后,使用 `np.arange()` 函数生成订单编号,使用 `np.random.randint()` 函数随机生成游乐项目编号和购买数量,最后计算订单金额并保存数据到文件。其中,`np.savetxt()` 函数用于将数据保存到 CSV 文件中,`delimiter=','` 表示使用逗号作为分隔符,`fmt='%d'` 表示数据格式为整数类型。