Skip to content

Instantly share code, notes, and snippets.

@oilsinwater
Created October 29, 2017 07:22
Show Gist options
  • Save oilsinwater/81eb5c62555bf0c5b6e243956c8eb825 to your computer and use it in GitHub Desktop.
Save oilsinwater/81eb5c62555bf0c5b6e243956c8eb825 to your computer and use it in GitHub Desktop.
csv to Python dict to Pandas DataFrame
import csv
from pandas import DataFrame as df
# -- View --
def transform_data(file_name):
"""Function transform CSV file data into Python Dictionary Object"""
with open(file_name, 'r') as file_handle:
key_list = []
value_list = []
data = csv.reader(file_handle, delimiter=',', quotechar='"')
for row in data:
if len(row) > 0:
key_list.append(row[0])
value_list.append(row[1])
dict_obj = dict(zip(key_list, value_list))
return dict_obj
# -- Model --
def main():
file_name = "quote.csv"
columns_obj = ["Symbol", "Price"]
dict_obj = transform_data(file_name)
df_obj = df(dict_obj.items(), columns=columns_obj)
df_obj['Number Shares'] = 100
df_obj['Value'] = df_obj['Number Shares'] * df_obj.Price.astype(float)
print(df_obj)
# -- Controller --
if __name__ == '__main__':
main()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment