Design function so variables get assigned values without large if/else block

Question

There must be a better way to do this:

def map_leds(self, i):
        if i[0] == 1:
            self.ledg_1_1.setVisible(True)
            self.ledr_1_1.setVisible(False)
        else:
            self.ledg_1_1.setVisible(False)
            self.ledr_1_1.setVisible(True)

###############################################

        if int(i[1]) == 1:
            self.ledg_1_2.setVisible(True)
            self.ledr_1_2.setVisible(False)
        else:
            self.ledg_1_2.setVisible(False)
            self.ledr_1_2.setVisible(True)

###############################################

        if int(i[2]) == 1:
            self.ledg_1_3.setVisible(True)
            self.ledr_1_3.setVisible(False)
        else:
            self.ledg_1_3.setVisible(False)
            self.ledr_1_3.setVisible(True)

###############################################

        if int(i[3]) == 1:
            self.ledg_2_1.setVisible(True)
            self.ledr_2_1.setVisible(False)
        else:
            self.ledg_2_1.setVisible(False)
            self.ledr_2_1.setVisible(True)

###############################################

        if int(i[4]) == 1:
            self.ledg_2_2.setVisible(True)
            self.ledr_2_2.setVisible(False)
        else:
            self.ledg_2_2.setVisible(False)
            self.ledr_2_2.setVisible(True)

###############################################

        if int(i[5]) == 1:
            self.ledg_2_3.setVisible(True)
            self.ledr_2_3.setVisible(False)
        else:
            self.ledg_2_3.setVisible(False)
            self.ledr_2_3.setVisible(True)

###############################################

        if int(i[6]) == 1:
            self.ledg_3_1.setVisible(True)
            self.ledr_3_1.setVisible(False)
        else:
            self.ledg_3_1.setVisible(False)
            self.ledr_3_1.setVisible(True)

###############################################

        if int(i[7]) == 1:
            self.ledg_3_2.setVisible(True)
            self.ledr_3_2.setVisible(False)
        else:
            self.ledg_3_2.setVisible(False)
            self.ledr_3_2.setVisible(True)

###############################################

if int(i[8]) == 1:
            self.ledg_3_3.setVisible(True)
            self.ledr_3_3.setVisible(False)
        else:
            self.ledg_3_3.setVisible(False)
            self.ledr_3_3.setVisible(True)

I have 9 LEDS, and I my function map_leds is fed an list of len 9 populated with 0s and 1s. If the first value is 1, the first LED should be green (this is a GUI, so an LED is actually a picture and I have to switch between both pictures, hence the "setVisible" function). If the first value is 0, then the LED (that should be displayed) is the red one. And so on for all the other values.

There must be a better way to actually do all this than have a huge if-else statement block, right?

Answer 1

You want to do three things:

• to use the indices
• to transform a flat vector into a 3x3 array
• to transform int values to boleans

First of all, initialize your led_r as an array:

self.led_r = []
for i in range(3):
    self.led_r[i] = [your_obj_constructor() for j in range(3)]

Now you can use a two-liner:

for x in range(9):
    self.led_r[x // 3][x % 3].setVisible(i[x] == 1)

Answer 2

Maybe separate them into smaller functions?

def hand_ledg_1_1 (val):
    if val == 1:
        print("green")
        self.ledg_1_1.setVisible(True)
        self.ledr_1_1.setVisible(False)
    else:
        print("red")
        self.ledg_1_1.setVisible(False)
        self.ledr_1_1.setVisible(True)

def hand_ledg_1_2 (val):
    if int(val) == 1:
        self.ledg_1_2.setVisible(True)
        self.ledr_1_2.setVisible(False)
    else:
        self.ledg_1_2.setVisible(False)
        self.ledr_1_2.setVisible(True)

def hand_ledg_1_3 (val):
    if int(val) == 1:
        self.ledg_1_3.setVisible(True)
        self.ledr_1_3.setVisible(False)
    else:
        self.ledg_1_3.setVisible(False)
        self.ledr_1_3.setVisible(True)

def hand_ledg_2_1 (val):
    if int(val) == 1:
        self.ledg_2_1.setVisible(True)
        self.ledr_2_1.setVisible(False)
    else:
        self.ledg_2_1.setVisible(False)
        self.ledr_2_1.setVisible(True)

def hand_ledg_2_2 (val):
    if int(val) == 1:
        self.ledg_2_2.setVisible(True)
        self.ledr_2_2.setVisible(False)
    else:
        self.ledg_2_2.setVisible(False)
        self.ledr_2_2.setVisible(True)

def hand_ledg_2_3 (val):
    if int(val) == 1:
        self.ledg_2_3.setVisible(True)
        self.ledr_2_3.setVisible(False)
    else:
        self.ledg_2_3.setVisible(False)
        self.ledr_2_3.setVisible(True)

def hand_ledg_3_1 (val):
    if int(val) == 1:
        self.ledg_3_1.setVisible(True)
        self.ledr_3_1.setVisible(False)
    else:
        self.ledg_3_1.setVisible(False)
        self.ledr_3_1.setVisible(True)

def hand_ledg_3_2 (val):
    if int(val) == 1:
        self.ledg_3_2.setVisible(True)
        self.ledr_3_2.setVisible(False)
    else:
        self.ledg_3_2.setVisible(False)
        self.ledr_3_2.setVisible(True)

def hand_ledg_3_3 (val):
    if int(i[8]) == 1:
        print("green1")
        self.ledg_3_3.setVisible(True)
        self.ledr_3_3.setVisible(False)
    else:
        print("red1")
        self.ledg_3_3.setVisible(False)
        self.ledr_3_3.setVisible(True)

def map_leds(self, i):
    hand_ledg_1_1(i[0])
    hand_ledg_1_2(i[1])
    hand_ledg_1_3(i[2])
    hand_ledg_2_1(i[3])
    hand_ledg_2_2(i[4])
    hand_ledg_2_3(i[5])
    hand_ledg_3_1(i[6])
    hand_ledg_3_2(i[7])
    hand_ledg_3_3(i[8])

It's easier to read at least