在使用scipy / numpy时,我确实获得了存储在numpy.ndarray
>>> a
array([[ 0.15555605, 0.51031528, 0.84580176, 0.06722675],
[ 0.60556045, 0.62721023, -0.48979983, -0.04152777],
[-0.78044785, 0.58837543, -0.21146041, -0.13568023],
[ 0. , 0. , 0. , 1. ]])
>>> print(a)
[[ 0.15555605 0.51031528 0.84580176 0.06722675]
[ 0.60556045 0.62721023 -0.48979983 -0.04152777]
[-0.78044785 0.58837543 -0.21146041 -0.13568023]
[ 0. 0. 0. 1. ]]
如何在一行上打印结果?
我已经检查过了:
>>> numpy.get_printoptions()
{'precision': 8, 'threshold': 1000, 'edgeitems': 3, 'linewidth': 75, 'suppress': False, 'nanstr': 'nan', 'infstr': 'inf', 'formatter': None}
但即使将linewidth
设置为1000也不会改变这一点。有没有办法更改该类型的显示格式?
是否也可以在每个数字之间添加逗号(如数组显示但没有周围的array(...)
)?
答案 0 :(得分:10)
要将numpy.tolist()
打印到一行,您可以将其转换为具有内置函数的列表import numpy as np
arr = np.array(((1, 2, 3), (4, 5, 6), (7, 8, 9)))
示例:
print(arr)
[[1, 2, 3]
[4, 5, 6]
[7, 8, 9]]
简单打印数组:
numpy.tolist()
与print(array.tolist())
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
比较:
class Kerneltype(Enum):
tube = 0
subspace_KDE = 1
deltashift = 2
dist_sens_via_mass_1 = 3
def aslist(self):
return [self.tube, self.subspace_KDE, self.deltashift, self.dist_sens_via_mass_1]
def fromint(self, int):
return self.aslist()[int]
答案 1 :(得分:3)
NumPy提供了一些自定义打印的方法,例如np.array2string
。
对于这个答案,我假设你有这样一个数组:
>>> import numpy as np
... arr = np.array([[ 0.15555605, 0.51031528, 0.84580176, 0.06722675],
... [ 0.60556045, 0.62721023, -0.48979983, -0.04152777],
... [-0.78044785, 0.58837543, -0.21146041, -0.13568023],
... [ 0. , 0. , 0. , 1. ]])
threshold
设置为np.inf
。,
作为分隔符,可以将separator
设置为','
。但是它没有删除换行符的选项,只是
max_line_width
,它给出了最内层维度在一行中打印的字符数。因此,当您设置max_line_width=np.inf
时它适用于1D阵列,但它不适用于ND阵列。幸运的是,它返回一个可以操作的字符串,例如删除所有换行符:
>>> np.array2string(arr, threshold=np.inf, max_line_width=np.inf, separator=',').replace('\n', '')
'[[ 0.15555605, 0.51031528, 0.84580176, 0.06722675], [ 0.60556045, 0.62721023,-0.48979983,-0.04152777], [-0.78044785, 0.58837543,-0.21146041,-0.13568023], [ 0. , 0. , 0. , 1. ]]'
或使用正则表达式删除所有空格:
>>> import re
>>> re.sub(r'\s+', '', np.array2string(arr, threshold=np.inf, max_line_width=np.inf, separator=','))
'[[0.15555605,0.51031528,0.84580176,0.06722675],[0.60556045,0.62721023,-0.48979983,-0.04152777],[-0.78044785,0.58837543,-0.21146041,-0.13568023],[0.,0.,0.,1.]]'
同意这些并不是真正的“短”,它们也比转换为list
.tolist()
然后转换为字符串慢,但它可能是一个很好的选择,特别是如果你想自定义打印结果而不创建(可能是巨大的)不必要的列表。