Now it can import and export from csv. Also a few other improvements in making lists.

Author: davidruffner

seatingchart/people5.xlsx

@@ -8,6 +8,7 @@
 """
 import random
 import numpy as np
+import pandas
 #Make classes 
 
 class Guest():
@@ -55,6 +56,26 @@ def __init__(self,guestnames=None):
             self.guests = [Guest(name) for name in guestnames]
         self.guestdict = {g.get_name():g for g in self.guests}
 
+    def fromExcel(self, excelfile):
+        peopledf = pandas.read_excel(excelfile)
+        #Read in all the names
+        allnames = set()
+        for column in peopledf[1:]:
+            for rownum in peopledf.index:
+                name = peopledf.iloc[rownum][column]
+                #print rownum, column, name
+                if name is np.nan:
+                    continue
+                allnames.add(name)
+        self.guests = [Guest(name) for name in allnames]
+        self.guestdict = {g.get_name():g for g in self.guests}
+
+        #Find all the friends
+        for rownum in peopledf.index:
+            friends = [str(name) for name in peopledf.iloc[rownum].values if name is not np.nan]
+            [self.guestdict[friend].set_friendnames(friends) for friend in friends]
+
+
     def insert_guest(self,guest):
         self.guests.append(guest)
         self.guestdict[guest.get_name()]=guest
@@ -179,6 +200,11 @@ def print_seatingchart(self):
             index = self.seatdict[n]
             print (n,index)
 
+    def to_excel(self, outfile='SeatingChart.xlsx'):
+        chartdf = pandas.DataFrame(self.chart)
+        chartdf.index = ["Table {0}".format(i) for i in chartdf.index]
+        chartdf.to_excel(outfile)
+
 
 class Organizer():
     def __init__(self,guestlist,seatingchart):

seatingchart/people6.xlsx

@@ -4,37 +4,12 @@
 import matplotlib.pyplot as plt
 import copy
 
-guests = ["Antonette",  
-          "Cinda",  
-          "Nathan",  
-          "Deidra", 
-          "Merna",  
-          "Renata",  
-          "Londa",  
-          "Lila",  
-          "Kathryne",  
-          "Lieselotte",  
-          "Teofila",  
-          "Hubert",  
-          "Isadora",  
-          "Elina",  
-          "Enola",  
-          "Emile",  
-          "Brendan",  
-          "Kit",  
-          "Deetta",  
-          "Jade"
-          ]
-
-guestlist = sc.GuestList(guestnames=guests)
-for i,g in enumerate(guests):
-    gn = guests.pop(i)
-    friends = random.sample(guests,10)
-    guests.insert(i,gn)
-
-    guestlist.guestdict[gn].set_friendnames(friends)
-
-nguests = len(guests)
+guestlist = sc.GuestList()
+guestlist.fromExcel('people6.xlsx')
+guestlist.print_friendlist()
+
+
+nguests = len(guestlist.guests)
 seatsper = 5
 
 chart = sc.SeatingChart(guestlist,nguests/seatsper,seatsper)
@@ -83,6 +58,10 @@
 plt.xlabel="step"
 plt.ylabel="friendships"
 
+org0.sc.to_excel('SeatingChart0.xlsx')
+org1.sc.to_excel('SeatingChart1.xlsx')
+
+
 plt.show()