实现一个函数,用于计算矩阵按行或按列的平均值。该函数接收一个二维数组(矩阵)和计算模式作为输入,返回对应的平均值列表。
输入描述:
输入包含两行:1. 第一行为一个二维数组(矩阵),数组中的元素可以是整数或浮点数2. 第二行为计算模式,字符串类型,值为 "row" 或 "column"- "row": 按行计算平均值- "column": 按列计算平均值


输出描述:
返回一个浮点数列表,表示按指定模式计算的平均值:- 如果模式为 "row",返回每行的平均值- 如果模式为 "column",返回每列的平均值
示例1

输入

[[1, 2, 3], [4, 5, 6]]
row

输出

[2.0, 5.0]

说明


示例2

输入

[[1, 2, 3], [4, 5, 6]]
column

输出

[2.5, 3.5, 4.5]

备注:
1. 对应的输入、输出已给出,您只用实现核心功能函数即可。2. 支持numpy、scipy、pandas、scikit-learn库。
加载中...