Pandas数据帧二维数组，转换为三维数组

Question

def dataGen(x):
    x1=x/2
    x2=x/3
    x3=x+x1
    return x1,x2,x3
output3DArray=data_2D_ArrayDf.apply(dataGen)

上面的代码旨在从二维数组创建三维数组。输入2D阵列形状为[10,10]，预期输出形状为[10,10,3]。当前输出仅为元组的2D 10,10数组

Answer 1

这是一个简单的解决方法

import numpy as np


def dataGen(a_2d_array):
    x, y = a_2d_array.shape
    output = np.zeros((x, y, 3))
    output[:, :, 0] = a_2d_array / 2
    output[:, :, 1] = a_2d_array / 3
    output[:, :, 2] = a_2d_array * 3 / 2
    return output


a_2d_array = np.random.rand(10, 10)
a_3d_array = dataGen(a_2d_array)
print(a_3d_array.shape)

# (10, 10, 3)