本文共 398 字,大约阅读时间需要 1 分钟。
import numpy as np
def get_mean(feature1, feature2):"""将feature1和feature2横向拼接,然后统计拼接后的ndarray中每列的均值"""# 将两个数组按水平方向拼接combined = np.hstack((feature1, feature2))# 计算每列的平均值,并返回结果return combined.mean(axis=1)
if name == "main":test_case = {'feature1': [[1, 2, 3, 4], [4, 3, 2, 1], [2, 3, 4, 5]],'feature2': [[1], [2], [3]]}result = get_mean(test_case['feature1'], test_case['feature2'])print(result)
转载地址:http://booyk.baihongyu.com/