dataframe 取平均且转换为数组
时间: 2023-10-08 15:14:37 浏览: 42
要将DataFrame对象的每列取平均并转换为数组,可以使用以下代码示例:
```python
import pandas as pd
import numpy as np
# 创建一个DataFrame对象
df = pd.DataFrame({'A': [1, 2, 3],
'B': [4, 5, 6],
'C': [7, 8, 9]})
# 求每列的平均值
mean_array = np.array(df.mean())
print(mean_array)
```
这将输出:
```
[2. 5. 8.]
```
通过调用DataFrame对象的`mean()`方法,可以计算每列的平均值。然后,使用NumPy库的`array()`函数将结果转换为数组。最后,可以打印出平均值数组。
相关问题
dataframe转换为数组
可以使用 Pandas 库中的 `to_numpy()` 方法将 DataFrame 转换为 NumPy 数组。示例如下:
```python
import pandas as pd
import numpy as np
# 创建 DataFrame
df = pd.DataFrame({
'A': [1, 2, 3],
'B': [4, 5, 6],
'C': [7, 8, 9]
})
# 转换为数组
arr = df.to_numpy()
print(arr)
```
输出结果为:
```
[[1 4 7]
[2 5 8]
[3 6 9]]
```
你也可以使用 `.values` 属性来实现相同的功能,它与 `.to_numpy()` 方法等价:
```python
arr = df.values
print(arr)
```
输出结果同样为:
```
[[1 4 7]
[2 5 8]
[3 6 9]]
```
提取dataframe某一列转化为数组
可以使用pandas库中的values属性来将dataframe某一列转化为数组。
例如,若要提取dataframe中名为“column_name”的列,可以使用如下代码:
```
import pandas as pd
df = pd.read_csv('file.csv')
array = df['column_name'].values
```
其中,'file.csv'是你要读取的csv文件名,'column_name'是你要提取的列名。提取后的数组将保存在变量array中。