python - 取 numpy 数组中列的平均值

Question

我已经使用 numpy 从 csv 文件中获取数据。 numpy 数组的尺寸为:100*20。我如何取列的平均值(比如 col 3,5,8)并用包含这 3 个 cols 平均值的新列替换它们

如果

   col3 = 1,2,3,4
   col5 = 2,3,4,8
   col8 = 3,4,5,6

然后我想删除这 3 列并插入一个新列，其中每个条目都包含这 3 列中值的平均值

我想插入一个新列:2,3,4,6，删除前 3 列，最终数组的维度为 100*28

是否有任何 numpy 函数可以做到这一点？

Answer 1

a = np.arange(100*30).reshape(100,30) # change this to your own data
cols = [2, 4, 7]                      # columns to calculate averages, i.e. 3,5,8
b = a[:, cols]                        # data of the cols
c = b.mean(axis=1)                    # average
a_no_b = np.delete(a, cols, axis=1)   # data without the cols
a_final = np.c_[a_no_b, c]            # final result