added country code to country name, continent name script

Author: mjhea0

33_country_code.py

@@ -0,0 +1,51 @@
+importcsv
+importsys
+importjson
+
+"""
+Example usage:
+
+$ python 33_country_code.py 33_sample_csv.csv 33_country_codes.json
+"""
+
+
+defget_data(csv_file, json_file):
+countryCodes= []
+countryNames= []
+continentNames= []
+withopen(csv_file, 'rt') asfile_one:
+reader=csv.reader(file_one)
+withopen(json_file) asfile_two:
+json_data=json.load(file_two)
+all_countries=json_data["country"]
+forcsv_rowinreader:
+forjson_rowinall_countries:
+ifcsv_row[0] ==json_row["countryCode"]:
+countryCodes.append(json_row["countryCode"])
+countryNames.append(json_row["countryName"])
+continentNames.append(json_row["continentName"])
+
+return [
+countryCodes,
+countryNames,
+continentNames
+ ]
+
+
+defwrite_data(array_of_arrays):
+withopen('data.csv', 'wt') ascsv_out:
+writer=csv.writer(csv_out)
+rows=zip(
+array_of_arrays[0],
+array_of_arrays[1],
+array_of_arrays[2]
+ )
+forrowinrows:
+writer.writerow(row)
+
+
+if__name__=='__main__':
+csv_file_name=sys.argv[1]
+json_file_name=sys.argv[2]
+data=get_data(csv_file_name, json_file_name)
+write_data(data)