pandas merge example

pandas_merge.py

from pandas import DataFrame, merge
import numpy as np

# generate array of random small integers
df1 = DataFrame(np.random.random_integers(5,size=(5,2)),columns=['a','b'])

# find unique values
uq = df1.a.unique()

# now generate two associated columns per unique value
df2 = DataFrame({
    'a': uq,
    'bar': np.random.randn(uq.size),
    'baz': np.random.randn(uq.size)
})

print df1
print df2

# merge on the shared column
print merge(df1, df2, on='a')