Python使用NumPy完整打印数组元素及参数配置

默认打印行为与问题

在使用NumPy时，如果数组容量较小，直接使用 print 函数可以输出全部元素。然而，当数组容量过大时，直接打印会导致输出被截断，中间部分显示为省略号（...）。这种折叠行为虽然节省了显示空间，但在程序调试阶段，往往需要查看完整的数据分布，因此必须找到一种方法将所有元素打印出来。

使用 set_printoptions 完整打印

要解决数组折叠问题，可以使用 np.set_printoptions() 函数。通过设置 threshold 参数，可以控制数组不被折叠的元素数量阈值。将 threshold 设置为 np.nan（或 'nan'），即可强制NumPy打印数组中的所有元素，无论其大小。

# 导入NumPy库
import numpy as np

# 设置threshold为np.nan，打印全部元素
np.set_printoptions(threshold=np.nan)

# 打印一个包含100个元素的数组
print(np.arange(100))

# 打印一个10x10的二维数组
print(np.arange(100).reshape(10, 10))

关键参数详解

np.set_printoptions() 提供了多个参数来精细控制输出格式。以下是几个常用参数的说明：

• threshold：控制数组不被折叠的数量阈值。当元素数量超过此值时，中间部分会被省略。
• precision：控制浮点数输出结果保留的精度位数。
• edgeitems：控制折叠后首尾显示的个数。
• formatter：用于自定义输出格式化，它是一个可调用的字典，其中 'all' 是一个常用的key。

precision 参数示例

通过设置 precision，可以控制浮点数的显示精度。例如，将精度设置为4：

np.set_printoptions(precision=4)
print(np.array([1.23456789]))
# 输出: [ 1.2346] # 注意最后进位了

threshold 与 edgeitems 参数示例

threshold 决定了何时触发折叠，而 edgeitems 决定了折叠时显示的首尾元素个数。

# 生成1到10的数组
arr = np.arange(1, 11, 1)

# 设置threshold为10，刚好不折叠
np.set_printoptions(threshold=10)
print(arr)
# 输出: [ 1 2 3 4 5 6 7 8 9 10]

# 设置threshold为9，触发折叠
np.set_printoptions(threshold=9)
print(arr)
# 输出: [ 1 2 3 ..., 8 9 10]

# 调整edgeitems为4，显示更多首尾元素
np.set_printoptions(threshold=5, edgeitems=4)
print(arr)
# 输出: [ 1 2 3 4 ..., 7 8 9 10]

formatter 自定义格式化

formatter 参数允许用户通过Lambda函数自定义每个元素的输出格式。例如，将每个整数转换为带有前缀的负数字符串：

np.set_printoptions(formatter={'all': lambda x: 'int: ' + str(-x)})
print(np.arange(1, 5, 1))
# 输出: [int: -1 int: -2 int: -3 int: -4]

重置打印选项

在完成调试或特定输出需求后，可以通过再次调用 np.set_printoptions() 并传入默认值或新的参数，来重置打印选项，恢复NumPy的默认行为。

# 重置为默认设置
np.set_printoptions()

以上就是Python使用NumPy完整打印数组元素及参数配置的详细内容，更多关于Python NumPy教程的资料请关注本站其它相关文章！