Align datasets

Shows how to combine information from two datasets describing the same cases, but not necessarily in the same order.

# Author: Christian Brodbeck <christianbrodbeck@nyu.edu>
import random
import string

from eelbrain import *


# Generate a dataset with known sequence
ds = Dataset()
ds['ascii'] = Factor(string.ascii_lowercase)
# Add an index variable to the dataset to later identify the cases
ds.index()

# Generate two shuffled copies of the dataset (and print them to confirm that
# they are shuffled)
ds1 = ds[random.sample(range(ds.n_cases), 15)]
ds1.head()
ascii index
w 22
u 20
n 13
d 3
i 8
s 18
p 15
h 7
r 17
e 4


ascii index
o 14
g 6
b 1
s 18
m 12
n 13
i 8
e 4
k 10
j 9


Align the datasets

Use the "index" variable added above to identify cases and align the two datasets

ds1_aligned, ds2_aligned = align(ds1, ds2, 'index')

# show the ascii sequences for the two datasets next to each other to
# demonstrate that they are aligned
ds1_aligned['ascii_ds2'] = ds2_aligned['ascii']
ds1_aligned
ascii index ascii_ds2
w 22 w
n 13 n
i 8 i
s 18 s
h 7 h
r 17 r
e 4 e
g 6 g
o 14 o


Gallery generated by Sphinx-Gallery