python保存二维数组到txt文件中的方法

以下是详细讲解“python保存二维数组到txt文件中的方法”的完整攻略。

1.准备工作

在保存二维数组到txt文件中之前，我们需要先导入Python的相关模块，如下所示：

import numpy as np

这里我们选择使用Python常用的科学计算库NumPy。

2.二维数组的创建

接下来，我们需要创建一个二维数组。我们可以通过NumPy的array函数来创建一个二维数组，如下所示：

a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

以上代码创建了一个包含3行3列的二维数组。我们可以通过print函数来打印该二维数组，如下所示：

print(a)

打印结果如下：

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

3.将二维数组保存到txt文件中

接下来，我们就可以将二维数组保存到txt文件中。我们可以使用NumPy中的savetxt函数来将数组保存到txt文件中，如下所示：

np.savetxt('array.txt', a)

以上代码将二维数组a保存到名为array.txt的txt文件中。我们可以通过以下代码来读取该txt文件中的数组：

b = np.loadtxt('array.txt')
print(b)

打印结果如下：

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

如果我们想要在txt文件中存储整数而不是浮点数，我们可以通过以下代码来设置数据格式：

np.savetxt('array.txt', a, fmt='%d')

以上代码将二维数组a保存到名为array.txt的txt文件中，并指定数据格式为整数。我们可以通过以下代码来读取该txt文件中的数组：

b = np.loadtxt('array.txt', dtype=int)
print(b)

打印结果如下：

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

4.示例说明

示例1

以下是一个示例程序，它创建一个包含3行4列随机整数的二维数组，并将其保存到名为random_array.txt的txt文件中：

import numpy as np

# 创建二维数组
a = np.random.randint(0, 10, (3, 4))

# 将二维数组保存到txt文件中
np.savetxt('random_array.txt', a, fmt='%d')

# 读取txt文件中的数组
b = np.loadtxt('random_array.txt', dtype=int)

# 打印结果
print('原始二维数组：')
print(a)
print('从txt文件中读取的二维数组：')
print(b)

输出结果如下：

原始二维数组：
[[6 8 0 3]
 [2 3 4 2]
 [5 5 2 9]]
从txt文件中读取的二维数组：
[[6 8 0 3]
 [2 3 4 2]
 [5 5 2 9]]

示例2

以下是另一个示例程序，它手动创建一个包含4行3列浮点数的二维数组，并将其保存到名为manual_array.txt的txt文件中：

import numpy as np

# 创建手动二维数组
a = np.array([[1.1, 2.2, 3.3], [4.4, 5.5, 6.6], [7.7, 8.8, 9.9], [0.1, 0.2, 0.3]])

# 将二维数组保存到txt文件中
np.savetxt('manual_array.txt', a)

# 读取txt文件中的数组
b = np.loadtxt('manual_array.txt')

# 打印结果
print('手动创建的二维数组：')
print(a)
print('从txt文件中读取的二维数组：')
print(b)

输出结果如下：

手动创建的二维数组：
[[1.1 2.2 3.3]
 [4.4 5.5 6.6]
 [7.7 8.8 9.9]
 [0.1 0.2 0.3]]
从txt文件中读取的二维数组：
[[1.1 2.2 3.3]
 [4.4 5.5 6.6]
 [7.7 8.8 9.9]
 [0.1 0.2 0.3]]

以上就是完整的“python保存二维数组到txt文件中的方法”的攻略及示例说明。感谢阅读！