Python NTuple.fill Examples

Example #1

File: test_ntuple.py Project: juanchopanza/pyhistuples

def test_access_row() :
    nt = NTuple('A','B')
    nt.fill('A', 5)
    rw = nt.row(0)
    ntl.assert_true(rw.A == 5)
    nt.fill('B',6)
    ntl.assert_true(rw.A==5 and rw.B == 6)

Example #2

def reference_ntuple():
    nt = NTuple('x', 'sin', 'cos')
    for x in xrange(-50, 51):
        nt.fill('x', x)
        nt.fill('sin', math.sin(x))
        nt.fill('cos', math.cos(x))
        nt.write()
    return nt

Example #3

File: test_ntuple.py Project: juanchopanza/pyhistuples

def test_get_with_cross_cut() :
    nt = NTuple('A', 'B')
    for x in range(100) :
        nt.fill('A', x)
        nt.fill('B', x*100)
        nt.write()
    col = nt.column('B',lambda x : x.A > 50 and x.B < 8500 )
    ntl.assert_true(col == [x*100 for x in xrange(51, 85)])

Example #4

File: test_ntuple.py Project: juanchopanza/pyhistuples

def test_get_with_cut() :
    nt = NTuple('A', 'B')
    for x in range(100) :
        nt.fill('A', x)
        nt.fill('B', x*100)
        nt.write()
    col = nt.column('A',lambda x : x.A > 50 )
    ntl.assert_true(col == range(51, 100))

Example #5

File: test_ntuple.py Project: juanchopanza/pyhistuples

def reference_ntuple() :
    nt = NTuple('x', 'sin', 'cos')
    for x in xrange(-50,51) :
        nt.fill('x',x)
        nt.fill('sin', math.sin(x))
        nt.fill('cos', math.cos(x))
        nt.write()
    return nt

Example #6

File: test_ntuple.py Project: juanchopanza/pyhistuples

def test_get_column() :
    nt = NTuple('A', 'B')
    for x in range(100) :
        nt.fill('A', x)
        nt.fill('B', x*100)
        nt.write()
    A = nt.column('A')
    ntl.assert_true(A == range(100))
    B = nt.column('B')
    ntl.assert_true(B == [ x*100 for x in range(100)])

Example #7

File: test_ntuple.py Project: juanchopanza/pyhistuples

def test_access_latest_row() :
    nt = NTuple('A', 'B')
    nt.fill('A', 5)
    rw = nt.row(0)
    ntl.assert_true(rw.A == 5)
    nt.fill('A', 7)
    rw = nt.row(0)
    ntl.assert_true(rw.A == 7)
    nt.write()
    nt.fill('A', 22)
    rw = nt.row(1)
    ntl.assert_true(rw.A == 22)

Example #8

File: test_pyhistoplots.py Project: juanchopanza/pyhistuples

__author__ = "Juan PALACIOS [email protected]"

from random import gauss
from pyhistuples.pyntuple.ntuple import NTuple
from pyhistuples.pyhistoplots import ntuple_plot, histo_plot, ntuple_column_histo

mu_p = 15.
mu_pt = 5.
sigma_p = 10.
sigma_pt = 5.

nt = NTuple('x', 'p', 'pt')

for x in xrange(10000):
    nt.fill('x', x)
    nt.fill('p', gauss(mu_p, sigma_p))
    nt.fill('pt', gauss(mu_pt, sigma_pt))
    nt.write()


def test_ntuple_plot():
    pt_plot = ntuple_plot(nt, 'pt', show=False)
    p_plot = ntuple_plot(nt,
                         'p',
                         errorfunction=lambda x: x.height / 2.,
                         color='red',
                         linewidth='1.5',
                         show=False)

Example #9

File: example_ntuple.py Project: juanchopanza/pyhistuples

import math
from random import gauss
from pyhistuples.pyntuple.ntuple import NTuple
from matplotlib import pyplot

mu_p = 15.
mu_pt = 5.
sigma_p = 10.
sigma_pt = 5.

nt = NTuple('x', 'p', 'pt')

for x in xrange(10000) :
    nt.fill('x',x)
    nt.fill('p', gauss(mu_p, sigma_p))
    nt.fill('pt', gauss(mu_pt, sigma_pt))
    nt.write()

x = nt.column('x', lambda row : row.p > 5. and row.pt > 1.)

Example #10

File: test_ntuple.py Project: juanchopanza/pyhistuples

def test_fill_column_twice() :
    nt = NTuple('A', 'B')
    nt.fill('A', 5)
    nt.fill('A', 7)

Example #11

File: test_ntuple.py Project: juanchopanza/pyhistuples

def test_illegal_tag_fails() :
    nt = NTuple('A','B')
    nt.fill('C', 5)

Example #12

File: test_ntuple.py Project: juanchopanza/pyhistuples

def test_fill_column() :
    nt = NTuple('A', 'B')
    nt.fill('A', 5)