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)]
print(ds1.head())
ds2 = ds[random.sample(range(ds.n_cases), 16)]
print(ds2.head())
ascii   index
-------------
s       18
f       5
y       24
x       23
l       11
i       8
p       15
b       1
e       4
v       21
ascii   index
-------------
y       24
f       5
i       8
v       21
u       20
l       11
d       3
r       17
b       1
m       12

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']
print(ds1_aligned)
ascii   index   ascii_ds2
-------------------------
f       5       f
y       24      y
l       11      l
i       8       i
b       1       b
v       21      v
a       0       a
w       22      w
k       10      k
z       25      z

Gallery generated by Sphinx-Gallery