示例#1
0
def root2np(tfile, tgraph):
    """ Converts ROOT data to numpy"""
    data = TFile(tfile).Get(tgraph)
    # From Root array to numpy array
    # see https://gist.github.com/jepio/4802f87164f20e266503
    x_data = np.array(data.GetX(), copy=True)
    y_data = np.array(data.GetY(), copy=True)
    data = np.vstack((x_data, y_data))
    return data