黑客|90个Numpy的有用的代码片段( 六 )


r[1:
= np.cumsum(dr) # integrate path
r_int = np.linspace(0 r.max() 200) # regular spaced path
x_int = np.interp(r_int r x) # integrate path
y_int = np.interp(r_int r y)
90 、 给定一个整数 n 和一个二维数组 X , 从 X 中选择可以解释为从具有 n 度的多项分布中抽取的行 , 即仅包含整数且总和为 n 的行、
# Author: Evgeni Burovski
X = np.asarray([[1.0 0.0 3.0 8.0

[2.0 0.0 1.0 1.0

[1.5 2.5 1.0 0.0

)n = 4
M = np.logical_and.reduce(np.mod(X 1) == 0 axis=-1)
M &= (X.sum(axis=-1) == n)
print(X[M
)
以上代码整理自:
【黑客|90个Numpy的有用的代码片段】https://avoid.overfit.cn/post/012966ac2b1848438801dcbd0d880c12