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

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

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)

用numpy处理数据即可,代码可这样写:

import numpy as np
#np.set_printoptions(suppress=True)
arr=np.random.randint(0,100,500,dtype=int).reshape(100,5)
arr1=arr.sum(axis=1)
np.savetxt('avg.csv',arr1,fmt='%d',encoding='gb2312')

如有帮助,请点采纳。

一行多少个数,一共多少列啊