matplotlib 版本 >=3 不包括 find()

Question

我正在运行一个非常简单的python script：

from tftb.generators import amgauss, fmlin

我收到此错误：

C:\Users\Anaconda3\envs\tf_gpu\lib\site-packages\tftb-0.0.1-py3.6.egg\tftb\processing\affine.py in <module>
     12 
     13 import numpy as np
---> 14 from matplotlib.mlab import find
     15 from scipy.signal import hilbert
     16 from scipy.optimize import brenth, newton

ImportError: cannot import name 'find'

我相信find不再是>=3 的版本。我怎样才能在不降级的情况下解决这个问题Matplotlib？

Answer 1

该matplotlib.mlab.find函数的代码是字面意思

import numpy as np

def find(condition):
    res, = np.nonzero(np.ravel(condition))
    return res

您可以用该功能替换任何出现的情况。