Numpy控制多维矩阵的显示格式

原文地址

分类目录——numpy

诸如长序列单行显示，全部显示（不缩略显示），精度（保留小数点后几位），是否科学计数法显示等等。

直接用程序来说明

• 生成测试数据

  import numpy as np
  
  np.random.seed(1)   # 控制每次生成的随机数据相同
  
  data = np.random.random(700).reshape((100,7))
  print(data)
  

  效果（部分）如下图

  

  可以看到默认设定，过长的会换行，默认进行科学计数法显示

• 通过numpy.set_printoptions()控制显示格式

  np.set_printoptions(
      infstr='inf',
      nanstr='nan',
      formatter=None,
      precision=2,    # 精度，保留小数点后几位
      threshold=500,
      # 最多可现实的Array元素个数
      # 限制的是基本元素个数，如3*5的矩阵，限制的是15而非3（行）
      # 如果超过就采用缩略显示
      edgeitems=3,
      # 在缩率显示时在起始和默认显示的元素个数
      linewidth=150,  # 每行最多显示的字符数，默认80，超过则换行显示
      suppress=True   # 浮点显示（不用科学计数法）
  )
  print(data)
  

  其中

  • precision=2

    精度，保留小数点后几位

  • threshold=500
    最多可现实的Array元素个数；
    限制的是基本元素个数，如3*5的矩阵，限制的是15而非3（行）;
    如果超过就采用缩略显示；
    设置inf则全部显示

  • edgeitems=3
    在缩略显示时在起始和默认显示的元素个数（对多个维度同时有效）

  • linewidth=150,

    每行最多显示的字符数，超过则换行显示

  • suppress=True

    浮点显示（不用科学计数法）

  如上设定之后显示如下

  

  应该注意这只是设置print出来的显示格式，并不会影响到数据在内存中运算的精度

• 相关文献

  Pytorch控制打印矩阵的格式