假设我们从这里开始
import numpy as np
a = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
如何将此有效地转换为 Pandas DataFrame,与以下相当:
import pandas as pd
>>> pd.DataFrame({'a': [0, 0, 1, 1], 'b': [1, 3, 5, 7], 'c': [2, 4, 6, 8]})
a b c
0 0 1 2
1 0 3 4
2 1 5 6
3 1 7 8
这个想法是让 a
列成为原始数组第一维的索引,其余列则是原始数组后两个维度中2D数组的垂直连接。
(这可以通过循环轻松完成; 问题是如何在没有循环的情况下完成。)
较长示例
使用 @Divakar 的卓越建议:
>>> np.random.randint(0,9,(4,3,2))
array([[[0, 6],
[6, 4],
[3, 4]],
[[5, 1],
[1, 3],
[6, 4]],
[[8, 0],
[2, 3],
[3, 1]],
[[2, 2],
[0, 0],
[6, 3]]])
应该将其改为类似于:
>>> pd.DataFrame({
'a': [0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3],
'b': [0, 6, 3, 5, 1, 6, 8, 2, 3, 2, 0, 6],
'c': [6, 4, 4, 1, 3, 4, 0, 3, 1, 2, 0, 3]})
a b c
0 0 0 6
1 0 6 4
2 0 3 4
3 1 5 1
4 1 1 3
5 1 6 4
6 2 8 0
7 2 2 3
8 2 3 1
9 3 2 2
10 3 0 0
11 3 6 3
'b': [1, 3, 5, 7]
吗?另外,你能否添加另一个样例,例如a = np.random.randint(0,9,(4,3,2))
,以便查看当维度长度不同时会发生什么? - Divakar