将六维数组重塑为二维数据帧
将六维数组重塑为二维数据帧
提问于 2017-02-06 05:06:26
我有一个六维数组A，我想把它改造成一个二维数组。结果矩阵的行应该通过A的前三个维度进行多索引，列应该通过A的最后三个维度进行多索引。使用pandas或numpy实现这一点的最佳方法是什么？
回答 1
Stack Overflow用户
发布于 2017-02-06 14:53:52
这里有一个很方便的函数来做这件事。
def make2d(a):
shape = a.shape
n = len(shape)
col_lvls = n // 2
idx_lvls = n - col_lvls
midx = pd.MultiIndex.from_product(
[range(i) for i in shape[:idx_lvls]],
names=['d-{}'.format(d) for d in range(1, idx_lvls + 1)])
mcol = pd.MultiIndex.from_product(
[range(i) for i in shape[idx_lvls:]],
names=['d-{}'.format(d) for d in range(idx_lvls + 1, idx_lvls + col_lvls + 1)])
return pd.DataFrame(
a.reshape(np.array(shape[:3]).prod(), -1),
midx, mcol
)演示
a = np.arange(216).reshape(2, 3, 2, 3, 2, 3)
make2d(a)
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/42057290
复制相关文章
相似问题
腾讯云开发者
