Pandas 從二維 arrays 的字典中創建 dataframe

Question

如何從二維 arrays 的字典中創建 dataframe 以使 dataframe 中的每個元素都是一個數組？ 例如，我希望下面的df.iloc[0]["A"]返回[0,0,0]

import pandas as pd
import numpy as np

data = {"A":np.zeros([100,3]), "B":np.zeros([100,3]), "C":np.zeros([100,3])}
df = pd.DataFrame.from_dict(data)
# print(df.iloc[0]["A"]) = [0,0,0]

Answer 1

我會使用 MultiIndex：

df = pd.DataFrame({k:x.ravel() for k,x in data.items()},
                  index=pd.MultiIndex.from_product([np.arange(100), np.arange(3)]))

然后df.loc[0,'A']給你一個系列：

0    0.0
1    0.0
2    0.0
Name: A, dtype: float64

Answer 2

如果你願意接受它們是 Python 列表而不是 np.arrays，你可以這樣做：

data = {"A": np.zeros([100,3]).tolist()}
df = pd.DataFrame.from_dict(data)

print(df.iloc[0]["A"])
[0.0, 0.0, 0.0]