import numpy as np
A = np.array([1,3,0,0,5])
B = np.array([2,0,0,4,3])
nzA = A!=0
nzB = B!=0
nzAB = nzA | nzB
nA = A[nzAB]
nB = B[nzAB]
nA
nB
* 如果想要提取两个向量对应坐标同时为零的坐标,只需将“ | ”改为“ & ”。
import numpy as np
A = np.array([1,3,0,0,5])
B = np.array([2,0,0,4,3])
nzA = A!=0
nzB = B!=0
nzAB = nzA | nzB
nA = A[nzAB]
nB = B[nzAB]
nA
nB
* 如果想要提取两个向量对应坐标同时为零的坐标,只需将“ | ”改为“ & ”。