我是numpy的新手。我有1000个3D结构的数据集,每个结构都有x,y,z坐标。我正在尝试使用CNN模型,但无法为该架构准备正确的数据集格式。这是我的熊猫数据框数据集:
row_id, xlist, ylist, zlist, target
1, [1,2,3,3], [5,2,6,7], [1,0,4,5], 1
2, [-1,6,4,6], [7,-1,4,1], [2,5,3,5], 1
3, [6,4,-2,-2], [2,-6,6,2], [3,0,7,2], 0
...
n, [5,3,1], [5,3,7], [3,5,3], 1
我尝试过,但这不是我想要的
np.array([dataset['xlist'],dataset['ylist'],dataset['zlist']])
我想获得具有x * y * z尺寸的3D结构。如何创建尺寸为4 * 4 * 4(用于x,y和z)的3D矩阵
答案 0 :(得分:1)
喀拉拉邦可能需要具有(高度,宽度,深度,通道)的表示形式。
因此,在您的情况下,可以先创建一个空白图像,如下所示:
empty_image = np.zeros(20, 20, 20, 1)
这假设您的x,y,z范围是-10到10,像素大小= 1 因此,对于每个点,您都可以将零变成一个:
empty_image[x+10, y+10, z+10, 0] += 1