numpy数组写入文件|极客教程

numpy数组写入文件

在数据处理和科学计算领域，NumPy是一个非常流行的Python库。NumPy提供了一个强大的数组对象，可以用来进行各种数学运算和操作。有时候我们需要将NumPy数组保存到文件中，以便之后进行读取和分析。本文将介绍如何将NumPy数组写入文件，以及一些常用的文件格式。

1. 将NumPy数组写入文本文件

最简单的方法是将NumPy数组写入文本文件。我们可以使用numpy.savetxt函数来实现这个功能。这个函数的语法如下：

numpy.savetxt(fname, X, fmt='%.18e', delimiter=' ', newline='\n', header='', footer='', comments='# ', encoding=None)

fname：要写入的文件名。
X：要写入的数组。
fmt：数据格式，可选参数，默认为'%.18e'。
delimiter：分隔符，可选参数，默认为空格。
newline：行分隔符，可选参数，默认为换行符。
header：文件头部内容，可选参数，默认为空。
footer：文件尾部内容，可选参数，默认为空。
comments：注释符号，默认为'#'。
encoding：文件编码，默认为None。

下面是一个示例，将一个二维NumPy数组写入文本文件：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
np.savetxt('data.txt', arr, fmt='%d', delimiter=',')

# 读取文件内容
with open('data.txt', 'r') as f:
    data = f.readlines()

print(data)

运行结果：

1,2,3
4,5,6
7,8,9

2. 将NumPy数组写入二进制文件

除了文本文件，我们还可以将NumPy数组写入二进制文件。这种方式更加高效，可以节省存储空间。我们可以使用numpy.save函数来实现这个功能。该函数的语法如下：

numpy.save(file, arr, allow_pickle=True, fix_imports=True)

file：要保存的文件名，以.npy为后缀。
arr：要保存的数组。
allow_pickle：是否允许保存Python对象，默认为True。
fix_imports：是否修复导入，一般设为True。

下面是一个示例：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
np.save('data.npy', arr)

# 读取文件内容
data = np.load('data.npy')

print(data)

运行结果：

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

3. 将NumPy数组写入CSV文件

CSV是一种常用的文本表格文件格式，逗号分隔值（Comma-Separated Values）。我们可以使用numpy.savetxt函数将NumPy数组保存为CSV文件。下面是一个示例：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
np.savetxt('data.csv', arr, fmt='%d', delimiter=',')

# 读取文件内容
with open('data.csv', 'r') as f:
    data = f.readlines()

print(data)