代码对numpy数组值求平均
时间: 2023-02-07 20:09:21 浏览: 67
可以使用 `numpy` 的 `mean` 函数来求数组的平均值。
```python
import numpy as np
arr = np.array([1, 2, 3, 4])
mean_val = np.mean(arr)
print(mean_val) # 输出 2.5
```
如果想在指定的轴上求平均值,可以在调用 `mean` 函数时传入 `axis` 参数。例如:
```python
arr = np.array([[1, 2], [3, 4]])
mean_val = np.mean(arr, axis=0) # 求每列的平均值
print(mean_val) # 输出 [2. 3.]
mean_val = np.mean(arr, axis=1) # 求每行的平均值
print(mean_val) # 输出 [1.5 3.5]
```
相关问题
numpy数组对某一维度取平均
可以使用numpy的mean函数对某一维度取平均。假设数组为`arr`,想要对第1维度取平均,则可以使用以下代码:
```python
import numpy as np
mean_arr = np.mean(arr, axis=0)
```
其中`axis=0`表示对第1维度取平均。如果要对其他维度取平均,只需要将`axis`参数的值改为相应的维度即可。
对一个n*m的2D numpy数组求平均值,获得1*m的1D数组
可以使用numpy的mean函数,设定axis参数为0来计算每列的平均值,得到一个1*m的1D数组。
示例代码如下:
```python
import numpy as np
# 生成一个n*m的2D numpy数组
n = 3
m = 4
arr = np.random.rand(n, m)
# 计算每列的平均值,得到一个1*m的1D数组
mean_arr = np.mean(arr, axis=0)
print(mean_arr)
```
输出:
```
[0.4153799 0.50885138 0.46834348 0.49067311]
```