如何将dataframe按行生成二维数组
时间: 2023-11-21 13:05:58 浏览: 123
python 使用pandas的dataframe一维数组和二维数组分别按行写入csv或excel
可以使用 `pandas` 库中的 `values` 属性将 `DataFrame` 转换为二维数组。默认情况下,`values` 属性返回一个包含所有行的二维数组,其中每一行都是一个子数组。以下是一个示例代码:
```python
import pandas as pd
# 创建一个DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
# 使用values属性将DataFrame转换为二维数组
array1 = df.values
# 输出转换后的二维数组
print(array1)
```
输出结果为:
```
[[1 4 7]
[2 5 8]
[3 6 9]]
```
这样,`DataFrame` 中的每一行都被转换为了二维数组中的一个子数组。如果需要按列生成二维数组,可以使用 `transpose()` 方法对转换后的数组进行转置。
阅读全文