任务详情:
给定一个DataFrame对象 df,要求返回各行的平均值。
具体操作如下:
-
添加新列 'avg' 用于计算各行的平均值;
-
使用数组返回新列 'avg'。
任务要求:
-
程序接收 DataFrame 对象 df,返回结果是 list 数据类型;
-
注意:平均值需要四舍五入保留两位小数。
思路:
1.使用Numpy中的mean()方法计算出DataFrame里每一行的平均值
axis=1/0 1是行,0是列
2.新建一个列表,遍历之前得到的数据,按照要求四舍五入保留两位小数,并添加到新的列表当中。返回列表即可。
代码如下:
import numpy as np
class Solution:
def CalAvg(self, df: 'pandas.DataFrame') -> list:
avgs=np.mean(df,axis=1).values
avg=list()
for x in avgs:
a=round(x,2)
avg.append(a)
return avg
pass
以上仅代表个人的思路,如有欠缺请指教。