python - pandas 数据帧到 numpy 数组，没有烦人的 dtype ="blah blah"

Question

我选择了第一列的几行，并使用 .values 和 as_matrix() 将其转换为 numpy 数组。

x1 = df.iloc[[1,3,4,64,34], 0]

数据帧中每个元素的维度是 (3, 3, 5) numpy 数组。我想从数据帧输出一个 (5, 3, 3, 5) numpy 数组。但是，当我使用 .values 时，我得到一个 (5, ) 形状数组，每个元素末尾都有 dtype =unit8 。我该如何解决这个问题？

Answer 1

df = pd.DataFrame(dict(A=[np.ones((3, 3, 5))] * 5))

A = np.array([a.tolist() for a in df.iloc[[0, 1, 2, 3, 4], 0]])

print(A.shape)

(5, 3, 3, 5)