使用Numpy在标准输出中写一个数组

使用Numpy在标准输出中写一个数组

在本文中,我们将介绍如何使用Numpy在Python中将一个数组写入标准输出。Numpy是Python中的一个常用库,它提供了大量用于处理多维数组的函数和操作。标准输出是指屏幕输出,也可以是重定向到文件中的输出。

阅读更多:Numpy 教程

创建一个数组

在使用Numpy将一个数组写入标准输出之前,我们首先需要创建一个数组。可以使用numpy.array()函数创建一个数组,并将其分配给一个变量。例如,以下代码创建一个包含5个整数的数组:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
Python

将数组写入标准输出

在Python中,可以使用print()函数将文本输出到标准输出。但是,如果要将数组写入标准输出,则需要使用特殊的Numpy函数numpy.savetxt()numpy.savetxt()函数可以接受三个参数:文件名、数组和分隔符。

但是,在不指定文件名的情况下,numpy.savetxt()函数将输出到标准输出。以下代码将上述数组写入标准输出:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
np.savetxt(sys.stdout, arr)
Python

输出结果如下:

1.000000000000000000e+00
2.000000000000000000e+00
3.000000000000000000e+00
4.000000000000000000e+00
5.000000000000000000e+00
Python

从输出结果可以看出,Numpy使用科学计数法将数组的每个元素写入标准输出。

如果要将元素按照指定格式打印,则需要传递一个格式字符串的参数:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
np.savetxt(sys.stdout, arr, fmt='%i')
Python

输出结果如下:

1
2
3
4
5
Python

在上面的示例中,我们使用%i作为格式字符串,指定将每个数组元素格式化为整数输出。

将多维数组写入标准输出

除了一维数组,Numpy还可以处理任意维度的数组。以下代码演示如何将二维数组写入标准输出:

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]])
np.savetxt(sys.stdout, arr, fmt='%i')
Python

输出结果如下:

1 2
3 4
5 6
Python

可以看到,Numpy以行作为单位将矩阵写入标准输出。

指定分隔符

默认情况下,numpy.savetxt()将使用空格分隔数组的元素。但是,您可以指定使用其他字符,例如逗号或制表符。

以下代码演示如何将逗号作为分隔符写入二维数组:

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]])
np.savetxt(sys.stdout, arr, fmt='%i', delimiter=',')
Python

输出结果如下:

1,2
3,4
5,6
Python

总结

在本文中,我们介绍了如何使用Numpy在Python中将一个数组写入标准输出。我们还介绍了如何指定分隔符和格式化选项。希望这篇文章能够帮助您使用Numpy编写更加高效的代码。

Python教程

Java教程

Web教程

数据库教程

图形图像教程

大数据教程

开发工具教程

计算机教程

登录

注册