[已解决]如何将一个矩阵，做成3D图

george1108

求诸位大佬：
例如：
[[0 1 2 3 4 5 6 7]
[1 0 3 2 5 4 7 6]
[2 3 0 1 6 7 4 5]
[3 2 1 0 7 6 5 4]
[4 5 6 7 0 1 2 3]
[5 4 7 6 1 0 3 2]
[6 7 4 5 2 3 0 1]
[7 6 5 4 3 2 1 0]]

上面的矩阵，想做成离散点的3D图。
x--列坐标，y--行坐标， z--对应矩阵值
(x, y, z)
(0, 0, 0)
(2, 7, 5)
(7, 5, 2)

x,y 我可以用linspace()，z的值该如何写？

最佳答案

月排行榜 / 总排行榜

学渣李某人

2021-7-10 17:42:36

1. import matplotlib.pyplot as plt
   
2. from mpl_toolkits.mplot3d import Axes3D
   
3. 
   
4. fig = plt.figure()
   
5. ax = fig.gca(projection='3d')
   
6. 
   
7. count = 100
   
8. range_ = 100
   
9. 
   
10. data = [[0, 1, 2, 3, 4, 5, 6, 7],
    
11.         [1, 0, 3, 2, 5, 4, 7, 6],
    
12.         [2, 3, 0, 1, 6, 7, 4, 5],
    
13.         [3, 2, 1, 0, 7, 6, 5, 4],
    
14.         [4, 5, 6, 7, 0, 1, 2, 3],
    
15.         [5, 4, 7, 6, 1, 0, 3, 2],
    
16.         [6, 7, 4, 5, 2, 3, 0, 1],
    
17.         [7, 6, 5, 4, 3, 2, 1, 0]]
    
18. 
    
19. x, y, z = [], [], []
    
20. 
    
21. for row in range(len(data)):
    
22.     for column in range(len(data[row])):
    
23.         x.append(row)
    
24.         y.append(column)
    
25.         z.append(data[row][column])
    
26. 
    
27. ax.scatter(x, y, z, s=z, c=z)
    
28. 
    
29. ax.set_xlabel('Row')
    
30. ax.set_ylabel('Col')
    
31. ax.set_zlabel('Value')
    
32. 
    
33. plt.show()
    

复制代码

跳转到最佳答案楼层

学渣李某人

试试这个代码:

1. import matplotlib.pyplot as plt
   
2. from mpl_toolkits.mplot3d import Axes3D
   
3. 
   
4. fig = plt.figure()
   
5. ax = fig.gca(projection='3d')
   
6. 
   
7. count = 100
   
8. range_ = 100
   
9. 
   
10. data = [[0, 1, 2, 3, 4, 5, 6, 7],
    
11.         [1, 0, 3, 2, 5, 4, 7, 6],
    
12.         [2, 3, 0, 1, 6, 7, 4, 5],
    
13.         [3, 2, 1, 0, 7, 6, 5, 4],
    
14.         [4, 5, 6, 7, 0, 1, 2, 3],
    
15.         [5, 4, 7, 6, 1, 0, 3, 2],
    
16.         [6, 7, 4, 5, 2, 3, 0, 1],
    
17.         [7, 6, 5, 4, 3, 2, 1, 0]]
    
18. 
    
19. x, y, z = [], [], []
    
20. 
    
21. for row in range(len(data)):
    
22.     for column in range(len(data[row])):
    
23.         x.append(row)
    
24.         y.append(column)
    
25.         z.append(data[row][column])
    
26. 
    
27. ax.scatter(x, y, z, s=z, c=z)
    
28. 
    
29. ax.set_xlabel('Row')
    
30. ax.set_ylabel('Col')
    
31. ax.set_zlabel('Value')
    
32. 
    
33. plt.show()

复制代码

原贴: https://paul.pub/matplotlib-3d-plotting/

学渣李某人

1. import matplotlib.pyplot as plt
   
2. from mpl_toolkits.mplot3d import Axes3D
   
3. 
   
4. fig = plt.figure()
   
5. ax = fig.gca(projection='3d')
   
6. 
   
7. count = 100
   
8. range_ = 100
   
9. 
   
10. data = [[0, 1, 2, 3, 4, 5, 6, 7],
    
11.         [1, 0, 3, 2, 5, 4, 7, 6],
    
12.         [2, 3, 0, 1, 6, 7, 4, 5],
    
13.         [3, 2, 1, 0, 7, 6, 5, 4],
    
14.         [4, 5, 6, 7, 0, 1, 2, 3],
    
15.         [5, 4, 7, 6, 1, 0, 3, 2],
    
16.         [6, 7, 4, 5, 2, 3, 0, 1],
    
17.         [7, 6, 5, 4, 3, 2, 1, 0]]
    
18. 
    
19. x, y, z = [], [], []
    
20. 
    
21. for row in range(len(data)):
    
22.     for column in range(len(data[row])):
    
23.         x.append(row)
    
24.         y.append(column)
    
25.         z.append(data[row][column])
    
26. 
    
27. ax.scatter(x, y, z, s=z, c=z)
    
28. 
    
29. ax.set_xlabel('Row')
    
30. ax.set_ylabel('Col')
    
31. ax.set_zlabel('Value')
    
32. 
    
33. plt.show()
    

复制代码

george1108

学渣李某人 发表于 2021-7-10 17:42

非常感谢大神回复！
测试了，就是我想要的！
再次感谢！

george1108

和大家分享一下运行的3D图。这好像是个立体分形

                               
登录/注册后可看大图

                               
登录/注册后可看大图

                               
登录/注册后可看大图