DataFrame 转化为数组
时间: 2023-08-14 13:39:20 浏览: 113
DataFrame 将某列数据转为数组的方法
5星 · 资源好评率100%
DataFrame 对象可以通过 values 属性转换为一个 numpy 数组。values 属性返回一个包含 DataFrame 中所有数据的 numpy 数组。例如,假设有以下 DataFrame 对象:
```
A B C
0 1 2 3
1 2 3 4
2 3 4 5
```
我们可以使用 values 属性将其转换为一个 numpy 数组:
```python
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B': [2, 3, 4], 'C': [3, 4, 5]})
array = df.values
```
这将返回一个 numpy 数组:
```python
array([[1, 2, 3],
[2, 3, 4],
[3, 4, 5]])
```
需要注意的是,转换后的 numpy 数组是按照 DataFrame 中的行列顺序组织的。即先按行展开,再按列展开。如果需要按照列的顺序展开,可以使用 numpy 库中的 transpose() 函数对数组进行转置操作:
```python
array = df.values.transpose()
```
这将返回一个按照列顺序展开的 numpy 数组:
```python
array([[1, 2, 3],
[2, 3, 4],
[3, 4, 5]])
```
阅读全文