python - 打印用于复制和粘贴的 numpy 数组

Question

有时，我想打印一个numpy数组，只是为了将它从命令行复制到其他地方。标准的 __repr__ 很接近，但缺少逗号:

import numpy as np

a = np.random.rand(5, 2)
print(a)

[[0.66585668 0.5793219 ]
 [0.28048686 0.11019737]
 [0.41359919 0.69354774]
 [0.02062253 0.85507001]
 [0.05443759 0.51366551]]

有什么提示吗？

Answer 1

假设您想要的输出如下:

[
    [0.5838947919313026, 0.05795999970550392],
    [0.9737053546477267, 0.7160755144818707],
    [0.17361370495727824, 0.09849691691484186], 
    [0.863423476768234, 0.7065666761268419],
    [0.18712559034118503, 0.21525050659592326]
]

您可以使用列表理解和 list 构造函数来执行此操作:

import numpy as np
a = np.random.rand(5, 2)
print([list(i) for i in a])