pandas 如何从熊猫中的两列创建一个数组

How to create an array from two columns in pandas

提问by Clement Attlee

suppose I have a DataFrame similar to this:

假设我有一个与此类似的 DataFrame：

d = {'col1': [0, 2, 4], 'col2': [1, 3, 5], 'col3': [2, 4, 8]}
df = pd.DataFrame(d)

   col1  col2  col3
0     0     1     2
1     2     3     4
2     4     5     8

How can I select col1 and col2 and turn them into this array?

如何选择 col1 和 col2 并将它们转换为这个数组？

array([[0, 1],
       [2, 3],
       [4, 5]])

回答by ayhan

You can access the underlying numpy array via the to_numpymethod:

您可以通过以下to_numpy方法访问底层的 numpy 数组：

df[['col1', 'col2']].to_numpy()
Out: 
array([[0, 1],
       [2, 3],
       [4, 5]])

---

---

.valuesattribute will do the same if you are on an earlier version (before v0.24).

.values如果您使用的是早期版本（v0.24 之前），则属性将执行相同的操作。

回答by Mohammad Akhtar

You can also achieve the same output with the below code.

您还可以使用以下代码实现相同的输出。

import numpy as np
np.array(df[['col1','col2']])
Out[60]: 
array([[0, 1],
       [2, 3],
       [4, 5]])