数据分析--将matplotlib图像转换为numpy.array

数据分析–将matplotlib图像转换为numpy.array

将 matplotlib.figure中的图片转换成array数组

import matplotlib.pyplot as plt
import numpy as np

def figure_to_array(myfig):
    """
    将 matplotlib.figure中的图片转换成array数组
    Args:
        myfig: figure对象
    Returns:
        numpy.ndarray
    """
    myfig.canvas.draw()
    X = np.array(myfig.canvas.renderer.buffer_rgba())
    return X