I would like to store some multidimensional data in a pandas dataframe or panel such that I would like to be able to return for example:
Simple approach:
runners = pd.DataFrame(
[
['Bob', 'Male', 1980],
['Tom', 'Male', 1986],
['Amy', 'Female', 1966],
],
columns=['Name', 'Gender', 'BirthYear']
)
races = pd.DataFrame(
[
['A', 2015, 'Bob', '2:35:09'],
['A', 2014, 'Bob', '2:47:34'],
['A', 2013, 'Bob', '2:50:12'],
['B', 2013, 'Bob', '1:32:07'],
['A', 2015, 'Tom', '3:05:56'],
['B', 1998, 'Amy', '1:29:43'],
],
columns=['Race', 'Year', 'Name', 'Time']
)
print races.loc[(races.Name == 'Bob') & (races.Race == 'A')][['Time']]
print
print races.loc[(races.Year == 2015) & (races.Race == 'A')][['Name', 'Time']]
Time
0 2:35:09
1 2:47:34
2 2:50:12
Name Time
0 Bob 2:35:09
4 Tom 3:05:56
get back all data
df = races.merge(runners)
get age at race time
df['Age'] = df.Year - df.BirthYear
print df
Race Year Name Time Gender BirthYear Age
0 A 2015 Bob 2:35:09 Male 1980 35
1 A 2014 Bob 2:47:34 Male 1980 34
2 A 2013 Bob 2:50:12 Male 1980 33
3 B 2013 Bob 1:32:07 Male 1980 33
4 A 2015 Tom 3:05:56 Male 1986 29
5 B 1998 Amy 1:29:43 Female 1966 32