一、什么是savetxt函数
savetxt函数是一个用于将numpy数组保存到文本文件中的函数。它可以将数组保存为多种格式，包括CSV、TSV等等，并且可以自定义数据的格式。
下面是一个采用默认设置来创建CSV文件的例子：
import numpy as np
arr = np.array([[1, 2], [3, 4]])
np.savetxt('my_array.csv', arr)
其中，’my_array.csv’是文件名，arr是numpy数组对象。默认情况下，savetxt函数将数组保存为CSV文件，并使用逗号作为分隔符。因此，文件内容如下：
1.000000000000000000e+00, 2.000000000000000000e+00 3.000000000000000000e+00, 4.000000000000000000e+00
二、如何将数据保存为不同格式
除了CSV格式之外，savetxt函数还可以将数据保存为其他格式，如TSV、TXT等。我们可以通过指定不同的分隔符来实现这个目的。
下面是一个采用制表符作为分隔符来创建TSV文件的例子：
import numpy as np
arr = np.array([[1, 2], [3, 4]])
np.savetxt('my_array.tsv', arr, delimiter='\t')
其中，delimiter参数用于指定分隔符，这里使用了制表符(‘\t’)。文件内容如下：
1.000000000000000000e+00 2.000000000000000000e+00 3.000000000000000000e+00 4.000000000000000000e+00
三、如何自定义数据的格式
除了保存数据为不同格式之外，我们还可以设置数据的输出格式。这可以通过fmt参数来实现。fmt参数是一个字符串，可以使用类似于C语言printf函数的占位符来指定数据的格式。
下面是一个采用自定义格式来创建TXT文件的例子：
import numpy as np
arr = np.array([[1, 2], [3, 4]])
np.savetxt('my_array.txt', arr, fmt='%d')
其中，fmt参数设置为’%d’，用于将数据格式化为整数。文件内容如下：
1 2 3 4
四、如何将多个数组存储到同一个文件中
savetxt函数还可以用于将多个数组存储到同一个文件中。我们可以将多个数组合并为一个二维数组，然后将其保存到文件中。
下面是一个将多个数组保存到同一个CSV文件中的例子：
import numpy as np
arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6], [7, 8]])
combined_arr = np.hstack((arr1, arr2))
np.savetxt('my_combined_array.csv', combined_arr)
其中，我们使用了numpy的hstack()函数将两个数组合并为一个。文件内容如下：
1.000000000000000000e+00, 2.000000000000000000e+00, 5.000000000000000000e+00, 6.000000000000000000e+00 3.000000000000000000e+00, 4.000000000000000000e+00, 7.000000000000000000e+00, 8.000000000000000000e+00
五、总结
savetxt函数是一个方便的numpy函数，可以将数组保存为多种格式的文本文件，并且可以自定义数据的格式。我们可以使用不同的分隔符、输出格式以及将多个数组保存到同一个文件中。
原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/231637.html
微信扫一扫 
支付宝扫一扫 