📜  将一组 numpy 数组保存到文件 py (1)

📅  最后修改于: 2023-12-03 14:53:46.545000             🧑  作者: Mango

将一组 numpy 数组保存到文件

当我们在处理数据时,有时需要将numpy的多个数组保存到文件中,以便后续的读取与处理,本文将介绍如何将一组numpy数组保存到文件。

具体实现

用numpy的savez函数可以将多个数组保存成一个npz文件,如下所示:

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.array([2, 4, 6, 8, 10])
c = np.array([3, 6, 9, 12, 15])

np.savez('array.npz', a=a, b=b, c=c)

这里将三个numpy数组abc保存到array.npz文件中。

如果需要将数组保存到.txt文件中,可以使用numpy.savetxt函数,如下所示:

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.array([2, 4, 6, 8, 10])
c = np.array([3, 6, 9, 12, 15])

np.savetxt("array.txt", np.c_[a,b,c], delimiter=', ', fmt='%d')

这里将三个numpy数组abc保存到array.txt文件中,使用np.c_将三个数组合并为一个数组,delimiter指定每个元素之间的分隔符,fmt指定元素的输出格式。

总结

本文介绍了如何使用numpy将多个数组保存到文件中,同时也介绍了numpy.savetext保存数组到文本文件的方法。将numpy数组保存到文件中,可以方便后续的读取与处理,尤其是在大型数据处理中,更能发挥numpy的效率。