从嵌套列表创建Panda DataFrame

Question

我试图从嵌套列表中创建一个包含ndarray的熊猫数据框：

from numpy import array
a = list([[1,2],[2,3]])                  
a[0] = array([[1,2]])
a[0][0] = array([1,2])

我想要实现的目标如下：

 D0    D1  
 1     2   
 2     3

我尝试过使用

pd.DataFrame(a)

创建

   D0      
 [1,2]        
 [2,3]     

我也尝试在for循环中使用pd.append

for i in range(0, len(a)):
  df = df.append(pd.DataFrame(a[i]))

实现了我想要的东西，但它非常慢，并且df.append会以某种方式创建重复项。

请帮忙。

提前谢谢。

Answer 1

pd.DataFrame构造函数直接接受列表列表。无需将列表元素重新定义为numpy数组。

a = [[1,2],[2,3]]

df = pd.DataFrame(a, columns=['D0', 'D1'])

print(df)

#    D0  D1
# 0   1   2
# 1   2   3