使用numpy生成一个100*5的数组A,要求每一列数据是随机生成的整数。求解数组A每一行的平均值,并将计算结果存储在100*1的数组B中。这里数组A第i行的平均值存储在数组B的第i个元素中。将计算结果存储在文件avg.csv中,要求采用GB2312编码
时间: 2024-11-05 11:24:48 浏览: 11
python numpy 一维数组转变为多维数组的实例
首先,我们需要安装`numpy`库,如果还没有安装,可以通过以下命令安装:
```bash
pip install numpy
```
然后,我们可以按照您的要求使用`numpy`生成数组,并计算每一行的平均值。下面是相应的Python代码:
```python
import numpy as np
# 生成一个100*5的数组,每列数据是随机整数(范围为0到100)
A = np.random.randint(0, 101, (100, 5))
# 计算每行的平均值并存储在一个新的100*1的数组B中
B = A.mean(axis=1)
# 将B数组的内容写入CSV文件,指定编码为GB2312
with open('avg.csv', 'w', encoding='GB2312') as f:
np.savetxt(f, B.reshape(-1, 1), fmt='%d', delimiter=',')
# 打印B数组内容(为了验证输出)
print(B)
```
这段代码首先生成了一个100行5列的随机整数矩阵,然后沿列方向计算平均值(`axis=1`),得到一个100行1列的新矩阵`B`。接下来,我们将这个结果保存到名为`avg.csv`的文件中,并设置编码为GB2312。
运行这段代码后,您会在当前目录下看到一个名为`avg.csv`的文件,里面包含的是B数组的平均值。如果您想检查每个元素是否正确,可以直接查看该文件的内容。
阅读全文