Question

我编写了一个脚本，用于在excel文档中搜索“ X”，当找到“ X”时，它将与“ X”关联的第一列和第一行复制到CSV文件中。

有人告诉我，有一种更好的方法来处理'if'语句。不太清楚如何。

代码如下：

 import xlrd
 import csv

 ###Grab the data from sheet 1
 def get_row_values(workSheet, row):
    to_return = []
     num_cells = myWorksheet.ncols - 1
    curr_cell = -1
    while curr_cell < num_cells:
        curr_cell += 1
        cell_value = myWorksheet.cell_value(row, curr_cell)
        to_return.append(cell_value)
     return to_return

 file_path = 'foo.xlsx'
 output = []

 #Write the data
 myWorkbook = xlrd.open_workbook(file_path)
 myWorksheet = myWorkbook.sheet_by_name('foosheet')
 num_rows = myWorksheet.nrows - 1
 curr_row = 0
 column_names = get_row_values(myWorksheet, curr_row)
 #print("TOTAL ENTRIES:") 
 #print len(column_names)
 #print("-----")
 framework_name = myWorksheet.cell(0,2)
 framework_version = myWorksheet.cell(0,3)
 while curr_row < num_rows:
        curr_row += 1 
        row = myWorksheet.row(curr_row)
        this_row = get_row_values(myWorksheet, curr_row)
        x = 0
        while x <len(this_row):
            if this_row[x] == 'x':
                output.append(['', fooname, foo_version, 
 foo_names[x], foo_row[0]])
                myData = [["foo1", "foo2", 
 "foo3", "foo4", "foo5"]]

                myFile = open('./results/barTemp.csv', 'w')
                with myFile:
                        writer = csv.writer(myFile)
                        writer.writerows(myData)
                        writer.writerows(output)

            x += 1

 #print output



 myFile.close()
 myWorkbook.release_resources()

Answer 1

不一定更好。运行时复杂度仍然相同。

区别将是更加紧凑的行：

例如，您可以更改

而x

到

对于this_row中的x：

但是我看到您使用'x'索引来查找column_names [x]，所以另一种方法可能更好，例如

对于x范围（len（this_row））：

如何在Python 2.7中将while语句重新编写为if语句？

1 个答案: