Python Connect-Four

I am currently using Python 3.5.2 to create a connect four basic game. My game should work for two players and tell you when you have won, be it horizontal vertical or diagonal. However I do not know how to make my code smaller or more efficient, is there a way to cut down, or is my method just not very efficient in itself?

I have tried looking up various other ways of doing this code, alas I found them difficult to understand. My code does not have any errors I have picked up on, and though I have completed the task I set myself, I do not feel this is the best way to do it.

If anybody had any suggestions, that would be much appreciated. If I knew how to make it shorter without losing functionality I would do so.

I hope my code will help anybody else attempting the same task as mine is rather basic.

```
import time

grid1 = [0,0,0,0] # bottom row
grid2 = [0,0,0,0]
grid3 = [0,0,0,0]
grid4 = [0,0,0,0]

grids = [grid1,grid2,grid3,grid4]

check = []

user = 1

class fullSlot_error (Exception):
pass
def hasWon_def():
print ("player "+str(user)+" has won")
time.sleep (1)

def grid_def():
print ("",grid4,"\n",grid3,"\n",grid2,"\n",grid1)

def user_def():
global user
if user < 2:
user = 2
else:
user = 1
return user

def slotFull_def():
while True:
try:
if grid4[userInput -1] != 0:
raise fullSlot_error
else:
break
except fullSlot_error:
print ("slot is full try again")
confirm_def()

def confirm_def():
looop= True
while looop== True:
try:
global userInput
userInput = int(input("\ninput a slot player "+str(user)+"(1,4)\n"))
if userInput < 5 and 0 < userInput:
looop = False
else:
print ("invalid int")
except ValueError:
print ("invalid input")

def placement_def():
counter = 0
for i in range (0,4):

Some suggestions:

• It is generally recommended to follow the pep8 style for Python code. For example no space between function names and paranthesis, lower_case_function_names, spaces after commas, etc.

• You can define your grid in a single operation rather than 5.

• For print, you can use * to unpack your nested list, then use sep='\n' to put each row on a different line.

• It would be better to have at least one top-level function, rather than doing everything in the root of the script.

• You don't need global variables, it is better to pass the variables as arguments and return the result.

• For print, if you give multiple values as arguments, spaces are automatically inserted between the arguments.

• You don't get any benefit from using a custom exception here, just use a break in your if test.

• you can use one-line if tests (called "ternary expressions") to simplify some of your code.

• There isn't much benefit from a simple one-line function that is only used in one place. It makes things harder to read and slows the code down a tiny bit. Just inline the code.

• There isn't much point using a sentinel variable (like looop) when you just assign False to it at a certain point. Just use break, and continue if need be to skip later parts of the loop.

• It is better to put the absolute minimal amount of code possible in your try block to avoid accidentally catching the wrong exception.

• you can combine multiple comparisons, such as 0






• You can use return instead of break to exit out of a while loop if you just want to return the result at that point with no further processing.






• You can use in to test if a value is in a list or other sequence of values rather then using and.






• You never use counter in placement_def.






• The 0 is implied in range(0, x), so you can leave it out.






• Rather than using range and indexing a list, it is easier to iterate over the list directly.






• You always subtract 1 from userInput. It would be easier to subtract one at the beginning.






• Rather than appending a list several times to another list, use extend to append the all the elements at once.






• check doesn't need to be global, its state is never shared between functions.






• You can use all to simplify checking rows or columns, and zip to transpose lists.






• You can use any and all to simplify your checks for whether a player has won and for whether the grid is full.






• You could simplify the code somewhat by having the check functions return a boolean, then printing in checks_def.






• I don't think placement_def is correct. You get a user input every time through the loop, rather than just once, and you overwrite the value if it is the current use or empty, when you want to overwrite only if empty. Further, you loop from top to bottom, when you want to loop from bottom to top.

• You hard-code the grid size, but it would be easy to allow the user to specify a square grid size.

So here is my code:

def play(n=None):
    if n is None:
        while True:
            try:
                n = int(input('Input the grid size: '))
            except ValueError:
                print('Invalid input')
                continue
            if n  user_input or user_input > n+1:
            print('invalid input:', user_input)
        elif grids[0][user_input-1] != 0:
            print('slot', user_input, 'is full try again')
        else:
            return user_input-1

def place_piece(user_input, user, grids):
    for grid in grids[::-1]:
        if not grid[user_input]:
            grid[user_input] = user
            return

def check_won(grids, user, n):
    return any(all(cell == user for cell in grid) for grid in grids)

def diagcheck_won(grids, user, n):
    return all(grids[x][x] == user for x in range(n))

if __name__ == '__main__':
    play()

Note that this could be simplified even further using numpy` arrays, but that is more advanced.

StackExchange Code Review Q#134636, answer score: 11