import matplotlib.pyplot as mp
import numpy as np
fig=mp.figure()
ax=fig.add_subplot(2,1,1)
mp.plot(data_3d[np.argwhere(mask==1)])
在上面的代码中，data_3d的形状为（32,32,4000），掩码的形状为（32,32）。将data_3d的（axis = 2）视为时间轴。最终，我想要一个沿x轴具有4000个刻度的图。而mask是一个32x32数组，其中每个元素为1或0。假设在1024个元素中，有300个元素是一个。我希望上面的代码绘制300条曲线。我意识到这可以通过for循环来实现。有没有更好的方法可以在一行代码中更快地完成它？