如何将nd numpy阵列打印(可视化)为2d矩阵?

2024-09-28 05:23:18 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个ndnumpy数组,它模拟一个2d矩阵,其中每个元素都是一个列向量。我需要打印二维数组,因为它是一个矩阵

以下是创建数组的代码:

Lx = 4
Ly = 4
f = 3
q = 10
pop = np.random.randint(q, size = (Lx, Ly, 1, f))

数组如下所示:

[[[3 3 6]]

[[5 2]]

[[6 0 2]]

[[5 8]]]

[[[7 8 2]]

我需要把它看作是这样的:

print(pop[0,0][0], pop[0,1][0], pop[0,2][0], pop[0,3][0])
print(pop[1,0][0], pop[1,1][0], pop[1,2][0], pop[1,3][0])
print(pop[2,0][0], pop[2,1][0], pop[2,2][0], pop[2,3][0])
print(pop[3,0][0], pop[3,1][0], pop[3,2][0], pop[3,3][0])

[2 9 2][5 5 8][4 9 3][4 2 3]

[3 9 0][6 4 0][8 3 9][6 4 1]

[4 8 9][2 2 6][1 3 8][4 5 6]

[4 1 5][8 0 9][3 9 7][4 4 3]

我不能使用示例中的print方法,因为数组可以更大。 理想情况下,将其可视化为本例中的图像将是惊人的:

Visualize 1D numpy array as 2D array with matplotlib

谢谢你的建议


Tags: 代码元素sizenply矩阵random数组
1条回答
网友
1楼 · 发布于 2024-09-28 05:23:18

删除维度的嵌套四循环:

In [46]: z = pop.reshape(tuple(d for d in pop.shape if d > 1))

In [47]: z.shape
Out[47]: (4, 4, 3)

In [48]: for x in z:
    ...:     for y in x:
    ...:         print(y, end=' ')
    ...:     print()
    ...:     
[4 0 3] [0 5 6] [7 4 2] [9 2 6] 
[8 3 1] [0 8 8] [5 7 9] [3 1 5] 
[4 9 9] [4 8 5] [0 8 7] [8 9 7] 
[8 7 4] [4 7 5] [1 8 5] [4 3 0] 

或者(使用相同的z):

In [55]: n = z.shape[1]

In [56]: fmt = '{} ' * n

In [57]: for x in z:
    ...:     print(fmt.format(*x))
    ...:     
[4 0 3] [0 5 6] [7 4 2] [9 2 6] 
[8 3 1] [0 8 8] [5 7 9] [3 1 5] 
[4 9 9] [4 8 5] [0 8 7] [8 9 7] 
[8 7 4] [4 7 5] [1 8 5] [4 3 0] 

相关问题 更多 >

    热门问题