举例:
绘制一个正方体
1 import matplotlib.pyplot as plt
2 from mpl_toolkits.mplot3d import Axes3D
3 import numpy as np
4 import nibabel as nib
5
6 # 创建一个图形对象
7 fig = plt.figure()
8
9 # 在图形中添加一个子图,参数为行数、列数和子图索引
10 ax = fig.add_subplot(1, 1, 1, projection='3d')
11
12 # 设置坐标轴的标签
13 ax.set_xlabel('X')
14 ax.set_ylabel('Y')
15 ax.set_zlabel('Z')
16
17 data = np.zeros((20, 20, 20))
18
19 data[2:15, 2:15, 2:15] = 1
20
21 # 创建一个新的Nifti1Image对象
22 solid_image = nib.Nifti1Image(data, np.eye(4))
23
24 # 保存为nii.gz文件
25 nib.save(solid_image, 'test.nii.gz')
26
27 # 绘制体素图
28 ax.voxels(data)
29
30 # 显示图形
31 plt.show()
显示结果: