Python中的 numpy.array_str()

numpy.array_str()是一个用于返回numpy.ndarray数组的字符串表示形式的函数。它可以接受多个参数，以控制输出格式，例如指定小数位数、转换数组形状等。

语法

numpy.array_str(arr, max_line_width=None, precision=None, suppress_small=None, separator=' ', prefix='', style=<no_value>, formatter=None)

参数说明：

• arr：numpy.ndarray数组
• max_line_width：每行的最大字符数，默认为None，即不限制
• precision：小数位数，默认为None，即不限制
• suppress_small：如果数组中有非常小的小数（比如1e-16），是否用0替换它，默认为None，即不替换
• separator：元素间的分隔符，默认为空格
• prefix：输出字符串的前缀
• style：输出的格式样式，默认为<no_value>，即不指定。可以为方括号、圆括号、大括号等
• formatter：一个函数，用于格式化数组元素。默认为None，即不作特殊处理。

返回值

返回指定格式的数组的字符串表示形式。

示例

import numpy as np

arr = np.array([[1, 2], [3, 4]])
print(np.array_str(arr))  # 输出：[[1 2]
                          #       [3 4]]

print(np.array_str(arr, precision=2, suppress_small=True))  # 输出：[[1.  2. ]
                                                            #       [3.  4. ]]

print(np.array_str(arr, max_line_width=10))  # 输出：[[1 2]
                                              #       [3 4]]

以上示例展示了如何使用numpy.array_str()函数来控制输出数组的格式形式。需要注意的是，numpy.array_str()函数返回的字符串不包含结尾的换行符，如果需要可以手动添加。