将列表与电子表格进行比较（Python / xlrd）

Question

I'm trying to compare two groups of account numbers (with Python and xlrd/ xlwt).

The first is a list of my favorite accounts. The seconds is a list of accounts that were recorded when someone from that account called me for help. However, the second list is actually a spreadsheet and has more than just the account numbers, such as a case ID and a category. For instance, account '100' called in about 'Shoes' and was recorded as case #50. (also assume that the spreadsheet has three columns: Account, Category, and Case #).

My objective is to look at the spreadsheet and find any of the times that someone from one of my favorite accounts (from the first list) called in for help. So I basically want to use something like

myFavoriteAccounts = ['100','200','300','400','500']

and then go through the entirety of the spreadsheet, printing any instance where one of my favorite accounts appears, as well as the case ID and the category.

I've been able to find the accounts that appear in both lists:

match = set(myFavoriteAccounts) & set(spreadsheetAccountsColumn)

But I don't know how to iterate through the spreadsheet and catch each time one of those accounts appeared as well as the category and case ID.

I'd like to be able to determine, for instance, that account '100' called in on two separate occasions about 'Shoes' for case #50 and then again for 'Socks' and case #70.

Answer 1

Assuming your data is a csv, you can use fileptr.readlines() to read it in, then split the lines based on your deliminator, from there it should be very easy to say

data = open('myfilepath','r').readlines()
data = [ d.split('delim') for d in data ]
accountitems = {}

for row in data:
    if row[0] in match: # the account number
        accountitems.setdefault(row[0],[]).append(line)

this will build you a dictionary whose keys are the account matches, and whose values are a list of all entries containing that account

you can also take a look at a modified bit of code I did using python csv, which may be helpful: http://code.activestate.com/recipes/577996-ordered-csv-read-write-with-colum-based-lookup/

or

import re
data = open('myfilepath','r').read() #note using read vs readlines
for fave in favelist:
    print "\n".join( re.findall(r"^%s.*$" % fave, data) ), "\n"

Answer 2

Here's some code as a skeleton.

xls = xlrd.open_workbook(xlsname)
worksheet = xls.sheet_by_name('Accounts') # Use whatever name is on the worksheet tab in excel
max_col = worksheet.ncols - 1 # Cells addressed from 0
max_row = worksheet.nrows - 1 # Cells addressed from 0
account_col = 0 # Assuming it's the first column

for row in xrange(0, max_row):
    value = xlrd.cell_value(row, account_col)
    if value in favorites:
        print "Do something"
        print "I can address other cells in this row if I want"
        for col in xrange(0, max_col):
            new_value = xlrd.cell_value(row, col)

I haven't tested this particular script, but I've used this method in my own programs.