将熊猫时间序列数据帧转换为3D数组

Question

我有一个带有两列“ USER_ID”和“ Daily_BALANCE”的时间索引数据帧。数据框显示了2018年4月一个月的用户每日余额（因此每个用户有30个观察结果）。我需要从该数据帧生成一个3D numpy数组，其形状（N（数据帧中的用户数Nb），30（每日日期），1（余额））保存相应的值。任何人都可以帮我实现此目的的代码。例如，如果我们在数据框中有100个用户，我得到一个形状为（100,30,1）的数组，该数组等于100个2D数组，每个数组有30行和一列。

谢谢

Answer 1

我假设您只希望将最终余额3D数组中的每日余额保留在其中，以便在ith日检索jth用户的余额，您可以执行{{1} }。

如果是这样，那么您首先需要按arr[i-1, j-1, 0]和USER_ID对数据框进行排序，然后获取numpy数组并最终对其进行DATE

reshape