featurize.py 文件源码

def print(self, f=sys.stdin):
        def pad(x):
            return x + " " * (4 - len(x))
        for r in range(self.mapHeight):
            s = ""
            for c in range(self.mapWidth):
                idx = r * self.mapWidth + c
                if self.terrain[idx] == Terrain.MOUNTAIN:
                    s += pad('M')
                elif self.terrain[idx] == Terrain.NEUTRAL_CITY:
                    s += pad('C')
                elif self.owner[idx] == Owner.NEUTRAL:
                    s += pad('.')
                elif self.owner[idx] == Owner.OURS:
                    s += pad(str(self.armies[idx]))
                else:
                    s += pad(str(-self.armies[idx]))
            print(s, file=f)
        print("move=%s" % str(self.nextMove), file=f)
        print("", file=f)