python中numpy数组的csv文件写入与读取

当我们在Python中使用Numpy库进行数据处理时，经常需要将Numpy数组保存到CSV文件中，或从CSV文件中读取Numpy数组。本文将详细介绍如何这两种操作。

Numpy数组写入CSV文件

在Numpy中，我们可以使用savetxt函数将Numpy数组保存到CSV文件中。下面一个示例，演示如何将Numpy数组保存到CSV文件中。

import numpy as np

# 创建一个Numpy数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8,9]])

# 将Numpy数组保存到CSV文件中
np.savetxt('data.csv', arr, delimiter=',')

print('保存成功！')

在上面的示例中，我们创建了一个Numpy数组arr，然后使用np.savetxt函数将其保存到CSV文件data.csv。delimiter参数指定了CSV文件中的分隔符，默认为逗号。输出结果为“保存成功！”。

需要注意的是，当将Numpy数组保存到CSV文件中时，CSV文件中的数据类型会被自动转换为字符串类型。如果需要保留原始数据类型，可以使用fmt参数指定数据格式。下面是一个示例，演示如何保始数据类型。

import numpy as np

# 创建一个Numpy数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 将Numpy数组保存到CSV文件中，并保留原始数据类型
np.savetxt('data.csv', arr, delimiter=',', fmt='%d')

print('保存！')

在上面的示例中，我们创建了一个Numpy数组arr，然后使用np.savetxt函数将其保存到CSV文件data.csv中，并使用fmt参数指定数据格式为整型。输出结果为“保存成功”。

从CSV文件中读取Numpy数组

在Numpy中，我们可以使用loadtxt函数从CSV文件中读取Numpy数组。下面是一个示例，演示如何从CSV文件中读取Numpy数组。

import numpy as np

# 从CSV文件中读取Numpy数组
arr = np.loadtxt('data.csv', delimiter=',')

print(arr)

在上面的例中，我们使用np.loadtxt函数从CSV文件data.csv中读取Numpy，并将其赋值给变量arr。输出结果如下：

[[1. 2. 3.]
 [4. 5. 6.]
 [7. 8. 9.]]

需要注意的，当我们从CSV文件中读取Numpy数组时，CSV文件中的数据类型会被自动换为浮点型。如果需要保留原始数据类型，可以使用dtype参数指定数据类型。下面是一个示例，演示如何保留原始数据类型。

import numpy as np

# 从CSV文件中读取Numpy数组，并保留原始数据类型
arr = np.loadtxt('data.csv', delimiter=',',=int)

print(arr)

在上面的示例中，我们使用np.loadtxt函数从CSV文件data.csv中读取Numpy数组，并使用dtype参数指定数据类型为整型。输出结果如下：

[[1 2 3]
 [4 5 6]
 [7 8 9]]

在这两个示例中，我们分别演示了如何将Numpy数组保存到CSV文件中，以及如何从CSV文件中读取Numpy数组。需要注意的是，在进行读写操作时，我们需要注意数据类型的一致性和分隔符的指定。