Commit fc124384 authored by Zolnierczuk, Piotr's avatar Zolnierczuk, Piotr
Browse files

bug in load_sqt_bin

parent 37e18e32

python/drspine/drspine.py

+4 −2
Original line number Diff line number Diff line
@@ -40,10 +40,11 @@ def _remove_empty_lines(lines, strip=True): 
            else:

                yield line



def load_txt(filename, delimiter=',', cleanup=None):

def load_txt(filename, delimiter=',', cleanup=None, skiprows=0):

    """wrapper around np.loadtxt """

    with open(filename) as fd:

        data = fd.readlines()

    data = data[skiprows:]

    # remove lines matching cleanup

    try:

        for remove in cleanup:
@@ -275,7 +276,8 @@ def plot_sqt(filename, **kwargs): 
#17    :  run_number

#18-20 :  q_vec

def _load_sqt_bin(filename, full_bw=False):

    s = load_txt(filename, delimiter=',', cleanup=['NaN', 'Infinity'])

    s = load_txt(filename, delimiter=',', cleanup=['NaN', 'Infinity'], skiprows=1)

    print(s.shape)



    sel = s[:,0]==0      # full bw data (it==0)

    if full_bw: