如何将只有一列数据的dataframe按行生成二维数组
时间: 2024-02-12 09:03:55 浏览: 72
Numpy将二维数组添加到空数组的实现
如果 `DataFrame` 只有一列数据,那么使用 `values` 属性转换后得到的是一个一维数组。如果需要将其转换为二维数组,可以使用 `reshape` 方法对其进行变形。以下是一个示例代码:
```python
import pandas as pd
# 创建一个只有一列的DataFrame
df = pd.DataFrame({'A': [1, 2, 3]})
# 使用values属性将DataFrame转换为一维数组
array1 = df.values.reshape(-1, 1)
# 输出转换后的二维数组
print(array1)
```
输出结果为:
```
[[1]
[2]
[3]]
```
这样,`DataFrame` 中的每一行都被转换为了二维数组中的一个子数组。`reshape` 方法中的参数 `-1` 表示自动计算数组的长度,参数 `1` 表示将一维数组变形为一列的二维数组。
阅读全文