将二维numpy数组储存为csv文件，同时设置储存的数据格式为整数

假设二维numpy数组为arr，可以使用numpy.savetxt()函数将其储存为csv文件，并设置数据格式为整数：

import numpy as np

# 生成测试数据
arr = np.random.rand(3, 4) * 100
arr = arr.astype(int)

# 储存为csv文件
np.savetxt('data.csv', arr, fmt='%d', delimiter=',')

其中，fmt参数设置数据格式为整数，delimiter参数设置分隔符为逗号。储存在data.csv文件中的数据格式如下：

92,1,32,74
86,79,86,31
13,28,17,76

相关问题

python批量对一维数组前一个和后一个数组成一个二维数组，并保存

以下是一个示例代码，可以批量将一维数组转换为二维数组并保存为CSV文件：

import numpy as np
import csv

# 生成示例数据
data = np.random.randint(1, 10, size=(10,))
print("原始数据：", data)

# 将前后两个元素组成二维数组
new_data = np.zeros((len(data)-1, 2))
for i in range(len(data)-1):
    new_data[i] = [data[i], data[i+1]]
print("新数据：\n", new_data)

# 保存为CSV文件
with open('output.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['col1', 'col2'])
    for row in new_data:
        writer.writerow(row)

这个代码将生成一个包含10个随机整数的一维数组，然后将前一个和后一个元素组成一个二维数组，并将结果保存为CSV文件。输出如下：

原始数据： [4 9 7 4 4 4 8 4 1 4]
新数据：
 [[4. 9.]
 [9. 7.]
 [7. 4.]
 [4. 4.]
 [4. 4.]
 [4. 8.]
 [8. 4.]
 [4. 1.]
 [1. 4.]]

输出的CSV文件内容如下：

col1,col2
4.0,9.0
9.0,7.0
7.0,4.0
4.0,4.0
4.0,4.0
4.0,8.0
8.0,4.0
4.0,1.0
1.0,4.0

注意，这里使用了numpy库来处理数组，同时使用了csv库来保存结果。如果需要处理更大的数据，可以考虑使用Pandas库来提高效率。

在numpy用array生成游乐园订单数据并保存数据

好的，以下是用numpy生成游乐园订单数据并保存数据的示例代码：

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'` 表示数据格式为整数类型。