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

  • Post category:Python

在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=',', dtype=int)

print(arr)

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

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

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