Question

import xlrd
fileWorkspace = '/Users/Bob/Desktop/'

wb1 = xlrd.open_workbook(fileWorkspace + 'Book1.xlsx')
sh1 = wb1.sheet_by_index(0)

colA,colB,colC,colD = [],[],[],[]
for a in range(3,sh1.nrows):
    colA.append(int(sh1.cell(a,0).value))
    colB.append(float(sh1.cell(a,1).value))
    colC.append(str(sh1.cell(a,2).value))
    colD.append(str(sh1.cell(a,3).value))
print(colA)
print(colB)   # Need to add 10 if indicated
print(colC)
print(colD)   # Adding 10 is indicated by "add10" 
print('\n')

update_colB_values = []
for i in range(3,sh1.nrows):    
    if sh1.cell(i,3).value == "add10":
        add = (sh1.cell(i,1).value) + 10
        update_colB_values.append(add)
print(update_colB_values)

我获取了旁边有“add10”的值的所有新值。如何创建一个数组，用新的替换那些旁边带有“add10”的值？

最初从excel文件中拉出ColB打印：

[32.0, 5.0, 12.0, 1.0, 87.0, 9.0]  # This is fine

更新后，我希望有一个更新列表的列表，因为已添加10.0。所以，新的colB应该是：

[42.0, 15.0, 12.0, 1.0, 87.0, 19.0]   # new_colB. This is what I want after the update.

截至目前的整个输出：

[0, 1, 2, 3, 4, 5]
[32.0, 5.0, 12.0, 1.0, 87.0, 9.0]
['', '', '', '', '', '']
['add10', 'add10', '', '', '', 'add10']


[42.0, 15.0, 19.0]

Excel文件图片： Excel File

Answer 1

现在，您只是将值附加到已更改的更新列表中。相反，您希望在更新需要添加的值时附加所有值。这是一个使用现有列和列表推导的表单：

colB = [32.0, 5.0, 12.0, 1.0, 87.0, 9.0]
colD = ['add10', 'add10', '', '', '', 'add10']

update_colB_values = [B + 10 if D == 'add10' else B for B, D in zip(colB, colD)]
# [42.0, 15.0, 12.0, 1.0, 87.0, 19.0]

您应该可以使用此行替换实例化和for循环。

应该注意的是，内置的zip是一个非常有用的功能，您可以在之前的代码中使用它。例如，它的基本行为是＆＃34;转移＆＃34;一个2D列表。

matrix = [['a1', 'a2', 'a3'],
          ['b1', 'b2', 'b3'],
          ['c1', 'c2', 'c3']]

for row in zip(*matrix):
    print(list(row))

# ['a1', 'b1', 'c1']
# ['a2', 'b2', 'c2']
# ['a3', 'b3', 'c3']

函数zip采用任意数量的位置参数，并将所有ith元素匹配在一起。只要其中一行耗尽，迭代就会结束，因此如果要填充，zip_longest（Python 3）或izip_longest（Python 2）来自itertools非常有用。

要使用现有代码执行此操作，您可以添加else语句，以便包含不要需要添加的元素。

update_colB_values = []
for i in range(3,sh1.nrows):    
    if sh1.cell(i,3).value == "add10":
        add = (sh1.cell(i,1).value) + 10
        update_colB_values.append(add)
    else:                                    # add
        value = sh1.cell(i,1).value          # these
        update_colB_values.append(value)     # lines

用新数字替换列表中的旧数字并创建新数组

1 个答案: