In [19]:
class Tree:
'''Class for a tree'''
def __init__(self, tree_id, species):
'''
Constructor of a tree.
@param tree_id ID of a tree
@param species Species of the tree
'''
self.id = tree_id
self.species = species
def getSpecies(self):
return(self.species)
class Forest:
'''Class for a forest'''
def __init__(self, tree):
'''
Constructor of a forest.
@param tree a tree class object
'''
self.trees = []
self.trees.append(tree)
def addTree(self, tree):
self.trees.append(tree)
def getTrees(self):
return(self.trees)
In [22]:
t = Tree(1, "Eiche")
f = Forest(t)
t = f.getTrees()
t[0].getSpecies()
f.addTree(Tree(2, "Buche"))
t = f.getTrees()
print(len(t))
print(t[0].getSpecies())
print(t[1].getSpecies())