Simple Logic Error in Python

Question

Why doesn't the 3rd row of the CSV output have "hola"?

The (if levels[a] == "3") should ensure it's set to "hola".

Thanks for any ideas or help!

#-*- coding: utf-8 -*-
import csv
levels = [["1"], ["2"], ["3"]]
def column1Logic(self):
    self.column1 = "logic worked"
    self.column1 = self.greeting
class Row(object):
    column1 = "name"
    greeting = "oh"
    def __init__(self, level, greeting):
        self.level = level
        self.greeting = greeting
    def rowEntry(self, level, greeting):
        column1Logic(self)
        lol = [[self.column1]]
        lol[0] = self.column1
        file_writer.writerow([o for o in lol])
with open("test.csv", "wb") as test_file:
    file_writer = csv.writer(test_file)
    for a in range(0, len(levels)):
        if levels[a] == "3":
            greeting = "hola"
            food = Row(levels[a], greeting)
            food.rowEntry(levels[a], greeting)
        else:
            greeting = "hi"
            food = Row(levels[a], greeting)
            food.rowEntry(levels[a], greeting)

Dušan Maďar · Accepted Answer · 2015-07-07 19:30:55Z

1

You are iterating over a list of lists. Therefore you should check if you get ['3'] instead of just '3'.

# -*- coding: utf-8 -*-
import csv
levels = [["1"], ["2"], ["3"]]
def column1Logic(self):
    self.column1 = "logic worked"
    self.column1 = self.greeting

class Row(object):
    column1 = "name"
    greeting = "oh"
    def __init__(self, level, greeting):
        self.level = level
        self.greeting = greeting
    def rowEntry(self, level, greeting):
        column1Logic(self)
        lol = [[self.column1]]
        lol[0] = self.column1
        file_writer.writerow([o for o in lol])

with open("test.csv", "wb") as test_file:
    file_writer = csv.writer(test_file)
    for a in range(0, len(levels)):
        print levels[a]
        if levels[a] == ['3']:  # instead of levels[a] == "3"
            greeting = "hola"
            food = Row(levels[a], greeting)
            food.rowEntry(levels[a], greeting)
        else:
            greeting = "hi"
            food = Row(levels[a], greeting)
            food.rowEntry(levels[a], greeting)

edited Jul 7, 2015 at 19:30

answered Jul 7, 2015 at 19:28

Dušan Maďar

10k6 gold badges58 silver badges72 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Bob Willaker Over a year ago

OH MY GOSH, THANK YOU SO MUCH!!!! to @ Khalil Ammour he commented it there in the fix (# instead of levels[a] == "3")

Dušan Maďar Over a year ago

@Bob Willaker yep, that is how you accept the answer

Collectives™ on Stack Overflow

Simple Logic Error in Python

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest