我有一个dataframe,它只有10行,我想将它的列‘A'变成一个10行1列的矩阵,请给我python代码
时间: 2023-11-22 18:53:42 浏览: 126
python的dataframe转换为多维矩阵的方法
可以使用`reshape`函数将列向量转化为矩阵,例如:
``` python
import pandas as pd
import numpy as np
# 创建一个包含10行3列数据的DataFrame
df = pd.DataFrame(np.random.randn(10, 3), columns=['A', 'B', 'C'])
# 将'A'列转化为10行1列的矩阵
matrix = df['A'].values.reshape((10,1))
print(matrix)
```
其中,`df['A']`为选取DataFrame中的'A'列,`.values`将其转化为NumPy数组,`reshape((10,1))`将其转化为10行1列的矩阵。
阅读全文