python - 平均连续的正数

Question

我有一个 numpy 矩阵，每一行都有正数和负数的组合。

我想创建一个新向量，它给出矩阵中一行中所有正数的平均值。

例如，如果这是矩阵:

[[1 2 3 -1]

[2 5 -6 5]]

我想用以下值创建向量:

[[2]

[4]]

最快的方法是什么？

总会有正数。

Answer 1

如果保证每行至少有一个正数(>=0)，您可以将负数(不包括0)转换为NaN 与 np.where然后使用 np.nanmean沿着行，像这样 -

np.nanmean(np.where(A>=0,A,np.nan),axis=1)

sample 运行-

In [69]: A
Out[69]: 
array([[ 2,  3, -6, -6, -4],
       [-5, -6, -1, -1,  3],
       [-8,  5, -7, -9, -9],
       [-3,  0,  7, -5, -6]])

In [70]: np.nanmean(np.where(A>=0,A,np.nan),axis=1)
Out[70]: array([ 2.5,  3. ,  5. ,  3.5])