从带有条件python numpy的数组中删除一个元素

Question

我有 3 个大小相同的数组 a 和 b

a= [1,20,0,3,9,9,9,55]
b= [1,5,6,9,9,9,9,55]

我需要从 a 中删除元素 0 并删除数组 b 中相同索引中的元素，如下所示：

    a= [1,20,3,9,9,9,55]
    b= [1,5,9,9,9,9,55]



import numpy as np
import math
a = np.array([1, 0,3])
b = np.array([5, 6,9])


c= np.vstack((a,b)).T
c= c[(c[:,0]<>0)]
k= c[:,0]
f= c[:,1]
r= math.sqrt(np.mean(np.power(((k - f) / k),2)))
r

我的代码正在运行，但我不喜欢它你还有其他建议吗？

Answer 1

如果你有 a、b、c 作为 numpy 数组

valid_idx = (a != 0)
b = b[valid_idx]
c = c[valid_idx]