MachineLearning/Basic Python/load_csv.py at master · falling-star/MachineLearning · GitHub

History

58 lines (45 loc) · 1.38 KB

Raw

from csv import reader

import os

# Load CSV data

def load_csv(filename):

dataset = list()

with open(os.getcwd() + '/' + filename, 'r') as file:

csv_reader = reader(file)

for row in csv_reader:

if not row:

continue

dataset.append(row)

return dataset

# verify CSV data

filename = 'data.csv'

dataset = load_csv(filename)

print('Loaded data file {0} with {1} rows and {2} columns').format(filename, len(dataset), len(dataset[0]))

print(dataset[0])

# str column to float column

def str_column_to_float(dataset, column):

for row in dataset:

row[column] = float(row[column].strip())

# verify

for i in range(len(dataset[0])):

str_column_to_float(dataset, i)

print(dataset[0])

# str column to float , int respectively

def str_column_to_int(dataset, column):

class_values = [row[column] for row in dataset]

unique = set(class_values)

lookup = dict()

for i, value in enumerate(unique):

lookup[value] = i

for row in dataset:

row[column] = lookup[row[column]]

return lookup

filename = 'iris.csv'

dataset = load_csv(filename)

print('Loaded data file {0} with {1} rows and {2} columns').format(filename, len(dataset), len(dataset[0]))

MachineLearning/Basic Python/load_csv.py at master · falling-star/MachineLearning · GitHub

Expand file tree

Breadcrumbs

load_csv.py

Latest commit

History

Breadcrumbs

load_csv.py

File metadata and controls