如何调整numpy数组元素之间的间距

Question

我想要这样类型的输出

[[ 1.  0.  0.]
[ 0.  1.  0.]
[ 0.  0.  1.]]

但我得到了这个

[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]]

我的代码是：

import numpy
print(numpy.identity(size))

Answer 1

这应该可以做到：

numpy.set_printoptions(formatter={'all': lambda x: " {:.0f} ".format(x)})

如果你想加小数点；｛：.0f｝"；通过"；｛：.0f｝；

您可以在lambda函数中修改空格数或格式。

Answer 2

按照使用numpy.arrange(开始、停止、大小(

import numpy as np
arr1 = np.array([[1,   2,   3,   4,   5], [6,  7,  8,  9,  0]])
arr2 = np.arange(1, 9, 1)
print(arr2)

输出

[1 2 3 4 5 6 7 8]