按行或列计算平均值

输入描述:

输入包含两行：
1. 第一行为一个二维数组（矩阵），数组中的元素可以是整数或浮点数
2. 第二行为计算模式，字符串类型，值为 "row" 或 "column"
- "row": 按行计算平均值
- "column": 按列计算平均值

输出描述:

返回一个浮点数列表，表示按指定模式计算的平均值：
- 如果模式为 "row"，返回每行的平均值
- 如果模式为 "column"，返回每列的平均值

备注:

1. Python3对应的输入、输出已给出，您只用实现核心功能函数即可。
2. 支持numpy、scipy、pandas、scikit-learn库。