datacamp/pandas_slicing_medals.py at master · 3210jr/datacamp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import pandas as pd

from glob import glob

bronze = pd.read_csv('bronze_top5.csv', index_col=0)
silver = pd.read_csv('silver_top5.csv', index_col=0)
gold = pd.read_csv('gold_top5.csv', index_col=0)


medal_types = ['bronze', 'silver', 'gold']

medals = []

for medal in medal_types:
    file_name = '%s_top5.csv' % medal
    columns = ['Country', medal]
    medal_df = pd.read_csv(file_name, header=0, index_col='Country', names=columns)
    medals.append(medal_df)

medals = pd.concat(medals, axis='columns', keys=medal_types)

print(medals)
"""

KeyError: 'the label [bronze] is not in the [index]'

medals_sorted = medals.sort_index(level=0)
print(medals_sorted.loc[('bronze','Germany')])
print(medals_sorted.loc['silver'])

idx = pd.IndexSlice

# Print all the data on medals won by the United Kingdom
print(medals_sorted.loc[idx[:, 'United Kingdom'], :])

"""

february = pd.concat(dataframes, axis=1, keys=['Hardware', 'Software', 'Service'] )

print(february.info())