numpy 打印固定大小的Python数组

vom3gejh  于 2023-05-07  发布在  Python
关注(0)|答案(1)|浏览(152)

是否可以显示固定大小的数组。例如,我的数组大小为28x82,但当我打印它时,它以13x60+4的格式显示
My code

ldfqzlk8

ldfqzlk81#

Numpy不再知道你的数组的大小,因为你已经重塑了它。你已经给了它形状60000, 784,然后得到了它的第二个元素,形状为784
如果你希望numpy将数组打印为28x28网格,你可以将形状指定为60000, 28, 28
此外,您可能希望指定numpy输出的最大线宽,以便数组的每一行实际占用一行输出,而不是溢出。

import numpy as np

...  # Define your array in the first place

np.set_printoptions(linewidth=200)  # Doesn't have to be 200, just anything that fits one row of the array
X_train.reshape(60000, 28, 28)[1]  # Reshape to 28x28

相关问题