python - 将数据框列转换为数组

Question

我正在尝试将我的数据框列转换为数组。例如，我有一个如下所示的数据框:

Total  Price   Carrier
2      3       C
1      5       D

我想像这样将列转换为数组:[[2, 1], [3,5], ['C','D]] 我不想要列名称。

我试过这样做:

df["all"] = 1
df.groupby("all")[["Total","Price", "Carrier"]].apply(list)

但是，我得到类似这样的东西 ["Total", "Price", "Carrier"] 并且是一个对象而不是数组。如何将所有列转换为数组？

Answer 1

使用df.values代替apply:

>>> df.values.T.tolist()
[[2, 1], [3, 5], ['C', 'D']]