在Python中,可以使用列表解析来实现将数组(2维)的每个内部数组的第i个元素相加,并将加法作为第i个元素构成一个新的数组。以下是实现的代码示例:
def add_arrays(arr):
result = [sum(sub_arr[i] for sub_arr in arr) for i in range(len(arr[0]))]
return result
这段代码中,arr
表示输入的二维数组。首先,使用列表解析遍历每个内部数组的第i个元素,并使用sum()
函数将其相加得到结果。然后,使用列表解析再次遍历每个内部数组的第i个元素构成新的数组result
。最后,返回结果数组result
。
这个方法适用于任意大小的二维数组,并且可以处理不同长度的内部数组。以下是一个示例的应用场景:
假设有一个二维数组arr
表示学生的成绩表,其中每个内部数组表示一个学生的成绩,第i个元素表示该学生的第i门课程的成绩。现在需要计算每门课程的平均成绩,可以使用上述代码来实现:
scores = [[80, 90, 70], [85, 95, 75], [90, 85, 80]]
averages = add_arrays(scores)
print(averages)
输出结果为:[85, 90, 75]
,表示每门课程的平均成绩分别为85、90和75。
推荐的腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云