在不使用 for 循环的情况下取另一个 numpy 数组中不同长度的 numpy arrays 的平均值

Question

我有一个包含其他 arrays 的二维数组，但 arrays 的大小不同。 我想在不使用循环的情况下找到该二维数组中包含的每个单独数组的平均值。

我想在下面的代码中取 array_combined 的平均值。

import numpy as np
array1=np.array([1,2,3,4,5])
array2=np.array([1,2,3])
array3=np.array([1,2,3,4,5])
list_combined = [array1,array2,array3]
array_combined = np.array(list_combined)

Answer 1

如果长度相等。

import numpy as np
list_combined=np.array([[1,2,3,4,5],[1,2,3,4,5],[1,2,3,4,5]])

print(list_combined.mean(axis=1))
#
''' 
array([3., 3., 3.])
'''
#
#

如果不

import numpy
import numpy as np
list_combined=np.array([[1,2,3,4,5],[1,2,3,numpy.NAN,numpy.NAN],[1,2,3,4,5]])

print(np.nanmean(list_combined,axis=1))
#
''' 
[3. 2. 3.]
'''
#
#

Answer 2

这应该可以解决您的问题：

import numpy as np
array1=np.array([1,2,3,4,5])
array2=np.array([1,2,3])
array3=np.array([1,2,3,4,5])
list_combined = [array1,array2,array3]
array_combined = np.array(list_combined)

mean_sub_arrays = np.vectorize(np.mean)(array_combined)

在 Google Colab 上运行它以验证并附上屏幕截图。 1个

从@vielkind 在此链接上提供的解决方案中复制： Average of elements in a subarray