Numpy：在特定数组位置计算表达式

Question

我有两个numpy数组dQ/dx和dQ/dt，我想计算v=（dQ/dt）/（dQ/dx），但只在dQ/dx和dQ/dt都不为零的位置。如果dQ/dx或dQ/dt等于零，则V=0。在示例中，dQ/dx=[0，0，0.2，0.1]，dQ/dt=[0.1，0，0.4，0]，应给出V=[0，0，2，0]。 我可以通过对所有数组元素进行循环来做到这一点，但是有没有更“numpy”的方法来做到这一点。先谢谢你。你知道吗

Answer 1

请尝试以下代码：

>>> import numpy as np
>>> dQdx = np.asarray([0, 0, 0.2, 0.1])
>>> dQdt = np.asarray([0.1 , 0, 0.4, 0])
>>> V = np.where(np.logical_and(dQdt, dQdx), dQdt/dQdx, 0)
>>> V
array([ 0.,  0.,  2.,  0.])