实现一个函数,用于计算矩阵按行或按列的平均值。该函数接收一个二维数组(矩阵)和计算模式作为输入,返回对应的平均值列表。
区块链毕设网qklbishe.com为您提供问题的解答
实现一个函数,用于计算矩阵按行或按列的平均值。该函数接收一个二维数组(矩阵)和计算模式作为输入,返回对应的平均值列表。
from typing import List, Union
import numpy as np
def calculate_matrix_mean(matrix: List[List[Union[int, float]]], mode: str) -> List[float]:
#首先按照对行或列求平均值进行分类讨论
if(mode == "row"):
ans = []
#如果是对行求平均值,则每次取出matrix中一行的数据,求平均值
for k in matrix:
sum = 0
for j in k:
sum += j
ans.append(1.0*sum/len(k))
return ans
else:
ans = []
#如果是对列求平均值,则每次取出matrix中一行列的数据,求平均值
for i in range(len(matrix[0])):
sum = 0
for j in range(len(matrix)):
sum+=matrix[j][i]
ans.append(1.0*sum/len(matrix))
return ans
def main():
matrix = eval(input())
mode = input()
result = calculate_matrix_mean(matrix, mode)
print(result)
if __name__ == "__main__":
main()
34:33
以上就是关于问题实现一个函数,用于计算矩阵按行或按列的平均值。该函数接收一个二维数组(矩阵)和计算模式作为输入,返回对应的平均值列表。的答案
欢迎关注区块链毕设网-
web3一级市场套利打新赚钱空投教程
区块链NFT链游项目方科学家脚本开发培训