find ship by Navy Fleet

Navy_Fleet.py

#!/usr/bin/env ipython
# author: superbil

# define values
map_size = 7
small_limit = 3
middle_limit = 2
bigger_limit = 1

# create map
import numpy as np
map = np.zeros((map_size, map_size), int)


# define ship
small_ship = 1
middle_ship = 2
bigger_ship = 4

# set default places(sea or ships)

# set prompt value for vertical
# get input from raw_input()
vertical = [4,0,0,1,1,1,3]           # example
#vertical = raw_input("vertical numbers:").split(' ')

# set prompt value for horizontal
horizontal = [0,4,1,1,2,1,1]         # example
#horizontal = raw_input("horizontal numbers:").split(' ')

# print "v:%s, h:%s" % (vertical, horizontal)           # debug use
                                     # get input from raw_input()
                                     # find ships
# find ship
# find a place to put ship on vertical
for i in range(len(vertical)):
    for j in range(len(horizontal)):
        # check v can put ship
        v = vertical[i]
        h = horizontal[j]
        if v > 0:
            print map[:v, j]

print map