python - 如何比较两个列表列表，然后在python中将数据写入excel

Question

我在 python 中有两个 list of lists，如下所示:

列表 1:

[['10001', 'Jason', 'University of Utah School of Medicine'],
 ['10007', 'Zhang', 'Duke'],
 ['10016', 'Som', 'Dana-Farber']]

列表 2:

[['Som', 'mrcc', 'Wed Dec 20 18:36:28 +0000 2017', '10016'],
 ['David', 'mrcc', 'Wed Dec 20 18:36:38 +0000 2017', '10013'],
 ['Kristen', 'mrcc gro', 'Wed Dec 20 18:37:24 +0000 2017', '10045'],
 ['Zhang', 'wan mrcc', 'Wed Dec 20 18:38:44 +0000 2017', '10007']]

现在我想根据比较这两个list of lists 的数据将数据写入excel 文件。我的 excel 文件的结构如下:

对于比较，我想做的是查看 list2 中每个列表的最后一个值是否出现在 list1 中的列表中。因此，对于上述示例列表，list2 中第一个列表中的 10016 出现在 list1 中。因此，如果它存在，那么我想在 excel 行中写入以下值:

[是，list1 的第三个列表中的值 2，list1 的第三个列表中的值 3，list2 的第一个列表中的值 2 , list2 的第一个列表中的值 3, list1 的第三个列表中的值 1]

如果 list2 中每个列表的最后一个值不存在于 list1 中，则将以下内容写入 excel 文件:

[No,N/A,N/A,list2当前列表中的值2, list2 的当前列表中的值 3，list2 的当前列表中的值 4]

我知道 python 中的 xlwt 等包用于写入 excel 文件 但我无法获得如何边写边做如上这样的比较？

更新:这是我目前的情况:

book = xlwt.Workbook(encoding="utf-8")

sheet1 = book.add_sheet("Sheet 1")


sheet1.write(0, 0, "User ID Present?")
sheet1.write(0, 1, "User Name")
sheet1.write(0, 2, "Center Name")
sheet1.write(0, 3, "Text")
sheet1.write(0, 4, "Date")
sheet1.write(0, 5, "User ID")

i=1
for items2 in list2:

    for items1 in list1:

        if items2[3] in items1[0]:
            sheet1.write(i, 0, "Yes")
            sheet1.write(i, 1, items1[1])
            sheet1.write(i, 2, items1[2])
            sheet1.write(i, 3, items2[1])
            sheet1.write(i, 4, items2[2])
            sheet1.write(i, 5, items1[0])
        else:
            sheet1.write(i, 0, "No")
            sheet1.write(i, 1, 'N/A')
            sheet1.write(i, 2, 'N/A')
            sheet1.write(i, 3, items2[1])
            sheet1.write(i, 4, items2[2])
            sheet1.write(i, 5, items2[3])
    i=i+1
book.save("Sample_Report_V1.xlsx")

当我运行这段代码时，它执行时没有错误，但是当我尝试打开文件时出现错误，因为文件由于错误的文件格式或文件扩展名而无法打开

list1 和 list2 是我上面描述的两个列表列表。

Answer 1

您想做的很简单，只是语法和组织代码的方式。这是我解决您的问题的方法:

import csv

def merge(list1, list2):
    lookup = {row[0]: row for row in list1}
    for row2 in list2:
        value = row2[-1]
        if value in lookup:
            row1 = lookup[value]
            yield 'Yes', row1[1], row1[2], row2[1], row2[2], row1[0]
        else:
            yield 'No', 'N/A', 'N/A', row2[1], row2[2], row2[3]

if __name__ == '__main__':
    list1 = [
        ['10001', 'Jason', 'University of Utah School of Medicine'],
        ['10007', 'Zhang', 'Duke'],
        ['10016', 'Som', 'Dana-Farber']]
    list2 = [
        ['Som', 'mrcc', 'Wed Dec 20 18:36:28 +0000 2017', '10016'],
        ['David', 'mrcc', 'Wed Dec 20 18:36:38 +0000 2017', '10013'],
        ['Kristen', 'mrcc gro', 'Wed Dec 20 18:37:24 +0000 2017', '10045'],
        ['Zhang', 'wan mrcc', 'Wed Dec 20 18:38:44 +0000 2017', '10007']]

    rows = merge(list1, list2)
    with open('merge_lists.csv', 'wb') as outfile:
        writer = csv.writer(outfile)
        writer.writerow(['User ID Present?', 'User Name', 'Center Name',
            'Text', 'Date', 'User ID'])
        writer.writerows(rows)

注意事项

• 您描述的过程在函数 merge 中。
• 为了便于查找，我创建了一个名为 lookup 的字典，其中用户 ID 是键，行是值。
• 写入 excel 文件，我猜你指的是 CSV 文件格式

更新

我的原始答案输出为 CSV 文件，我认为这是最容易处理的。您可以毫无困难地在 Excel 中打开该文件。但是，我有种感觉，你是非要写成Excel格式，所以我花了一些时间学习了xlwt，足以使用它了。这是我更新的解决方案。请注意，此模块处理较旧的 .xls 文件格式，而不是较新的 .xlsx 格式。如果保存为 .xlsx，Excel 将无法打开它。

import xlwt

def merge(list1, list2):
    lookup = {row[0]: row for row in list1}
    for row2 in list2:
        value = row2[-1]
        if value in lookup:
            row1 = lookup[value]
            yield 'Yes', row1[1], row1[2], row2[1], row2[2], row1[0]
        else:
            yield 'No', 'N/A', 'N/A', row2[1], row2[2], row2[3]

def write_row(sheet, row, row_number, starting_column_number=0):
    for column_number, cell in enumerate(row, starting_column_number):
        sheet.write(row_number, column_number, cell)

if __name__ == '__main__':
    list1 = [
        ['10001', 'Jason', 'University of Utah School of Medicine'],
        ['10007', 'Zhang', 'Duke'],
        ['10016', 'Som', 'Dana-Farber']]
    list2 = [
        ['Som', 'mrcc', 'Wed Dec 20 18:36:28 +0000 2017', '10016'],
        ['David', 'mrcc', 'Wed Dec 20 18:36:38 +0000 2017', '10013'],
        ['Kristen', 'mrcc gro', 'Wed Dec 20 18:37:24 +0000 2017', '10045'],
        ['Zhang', 'wan mrcc', 'Wed Dec 20 18:38:44 +0000 2017', '10007']]

    rows = merge(list1, list2)
    workbook = xlwt.Workbook(encoding='utf-8')
    worksheet = workbook.add_sheet('Sheet 1')

    write_row(
        worksheet,
        'User ID Present?,User Name,Center Name,Text,Date,User ID'.split(','),
        row_number=0)

    for row_number, row in enumerate(rows, 1):
        write_row(worksheet, row, row_number)

    workbook.save('Sample_Report_V1.xls')