问题描述:取数组中每n个元素的平均值,并将MATLAB代码转换为Python。
回答内容: 这个问题可以通过编写一个函数来解决,函数接受两个参数:数组和n,返回一个新的数组,其中每个元素是原数组中每n个元素的平均值。
MATLAB代码示例:
function result = average_array_elements(array, n)
num_elements = floor(length(array) / n);
result = zeros(1, num_elements);
for i = 1:num_elements
start_index = (i-1)*n + 1;
end_index = i*n;
result(i) = mean(array(start_index:end_index));
end
end
Python代码示例:
import numpy as np
def average_array_elements(array, n):
num_elements = len(array) // n
result = np.zeros(num_elements)
for i in range(num_elements):
start_index = (i-1)*n + 1
end_index = i*n
result[i] = np.mean(array[start_index:end_index])
return result
这个函数首先计算出结果数组的长度(取整数除法),然后创建一个全零数组来存储结果。接下来,使用循环遍历每个结果元素的索引,通过起始和结束索引来切片原始数组,并计算平均值,将结果存储在结果数组中。最后返回结果数组。
MATLAB代码和Python代码的实现思路相似,但是在细节上有些差异。在MATLAB中,可以使用数组的切片和mean()函数来实现,而在Python中,可以使用NumPy库的切片和mean()函数来实现。
领取专属 10元无门槛券
手把手带您无忧上云