数组A第i行的平均值存储在数组B的第i个元素中

Question

使用numpy生成一个1005的数组A，要求每一列数据是随机生成的整数、求解数组A每一行的平均值，并将计算结果存储再1001的数组B中。这里数组A第i行的平均值存储在数组B的第i个元素中。将计算结果存储再文件avg.csv中，要求采用GB2312编码。

Answer 1

import numpy as np
#使用numpy生成一个100*5的数组A，要求每一列数据是随机生成的整数、
dataA = np.random.randint(1,100,(100,5))
print(dataA)

求解数组A每一行的平均值，

mean = np.mean(dataA,axis=1)
print("数组A每一行的平均值:\n",mean)

并将计算结果存储再100*1的数组B中。

dataB = np.array([mean])
dataB.shape = 100,1
print("100*1的数组B:\n",dataB)