题解 | 按行或列计算平均值

from typing import List, Union
import numpy as np

def calculate_matrix_mean(matrix: List[List[Union[int, float]]], mode: str) -> List[float]:
    mat=np.array(matrix)
    if mode=="row":
        mean1=mat.mean(axis=1)
    else:
        mean1=mat.mean(axis=0)
    return mean1.tolist()
def main():
    matrix = eval(input())
    mode = input()
    result = calculate_matrix_mean(matrix, mode)
    print(result)

if __name__ == "__main__":
    main()