如何按列而不是按行将我的 dataframe 变成 numpy 数组？

Question

考虑到我有这个例子 dataframe：

d = {'Gender': [1,1,0,1,0], 'Employed': [1,0,0,1,1]}

我希望这是一个这种形式的数组：

[[1 1 0 1 0][1 0 0 1 1]]

当我跑

d[['Gender', 'Employed']].to_numpy()

我得到一个 [[1 1][1 0][0 0][1 1][0 1]] 形式的数组

Answer 1

您可以转置 DataFrame，使用values方法获取底层 numpy 数组并使用tolist方法转换为列表：

out = pd.DataFrame(d).T.values.tolist()

Output：

[[1, 1, 0, 1, 0],
 [1, 0, 0, 1, 1]]

Answer 2

这可以得到你预期的 output

import pandas as pd
df = pd.DataFrame({'Gender': [1,1,0,1,0], 'Employed': [1,0,0,1,1]})
print([df[column].to_list() for column in df])

Output：

[[1, 1, 0, 1, 0], [1, 0, 0, 1, 1]]

如果您对这里有更多期望，请告诉我。

Answer 3

转一下就行了。

d[['Gender', 'Employed']].values.T