Added Create Dataframe with Pandas in Machine Learning (hacktoberfest17#1176)

propersam · t2013anurag · commit 9eaf5474e5ea · 2017-10-24T20:01:49.000+05:30
diff --git a/machine_learning/Pandas_Dataframe/Create_DataFrame.py b/machine_learning/Pandas_Dataframe/Create_DataFrame.py
@@ -0,0 +1,50 @@
+from pandas import DataFrame, Series
+
+#################
+# Syntax Reminder:
+#
+# The following code would create a two-column pandas DataFrame
+# named df with columns labeled 'name' and 'age':
+#
+# people = ['Sarah', 'Mike', 'Chrisna']
+# ages  =  [28, 32, 25]
+# df = DataFrame({'name' : Series(people),
+#                 'age'  : Series(ages)})
+
+def create_dataframe():
+    '''
+    Create a pandas dataframe called 'olympic_medal_counts_df' containing
+    the data from the table of 2014 Sochi winter olympics medal counts.  
+
+    The columns for this dataframe should be called 
+    'country_name', 'gold', 'silver', and 'bronze'.  
+
+    There is no need to  specify row indexes for this dataframe 
+    (in this case, the rows will automatically be assigned numbered indexes).
+    
+    You do not need to call the function in your code when running it in the
+    browser - the grader will do that automatically when you submit or test it.
+    '''
+
+    countries = ['Russian Fed.', 'Norway', 'Canada', 'United States',
+                 'Netherlands', 'Germany', 'Switzerland', 'Belarus',
+                 'Austria', 'France', 'Poland', 'China', 'Korea', 
+                 'Sweden', 'Czech Republic', 'Slovenia', 'Japan',
+                 'Finland', 'Great Britain', 'Ukraine', 'Slovakia',
+                 'Italy', 'Latvia', 'Australia', 'Croatia', 'Kazakhstan']
+
+    gold = [13, 11, 10, 9, 8, 8, 6, 5, 4, 4, 4, 3, 3, 2, 2, 2, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0]
+    silver = [11, 5, 10, 7, 7, 6, 3, 0, 8, 4, 1, 4, 3, 7, 4, 2, 4, 3, 1, 0, 0, 2, 2, 2, 1, 0]
+    bronze = [9, 10, 5, 12, 9, 5, 2, 1, 5, 7, 1, 2, 2, 6, 2, 4, 3, 1, 2, 1, 0, 6, 2, 1, 0, 1]
+    
+    # your code here
+
+    ## My code starts here
+    data = {'country_name':Series(countries),
+            'gold':Series(gold), 'silver':Series(silver), 'bronze':Series(bronze)}
+    olympic_medal_counts_df = DataFrame(data)
+    #My code ends here
+    
+    return olympic_medal_counts_df
+
+print(create_dataframe())
