这是我在Excel工作表中查找特定值位置的函数。Uncool List Comprehension
from xlwings import Workbook, Range
workbook = Workbook('workbook location')
def get_data_locations():
""" Find data locations from the first sheet in document. """
first_sheet = Range('Sheet1', 'A1:Z200').value
coordinates = []
for x in range(len(first_sheet)):
coordinate_pair = [[x + 1, y + 1] for y, z in enumerate(first_sheet[x]) if z == 'CO']
if not coordinate_pair == []:
coordinates.append(coordinate_pair)
coordinates = [cp for [cp] in coordinates]
print(coordinates)
就我所见,代码按预期工作。然而,出于某种原因,我觉得我在这里杀死小狗。
例如,在这里添加嵌套列表似乎是多余的。
[x + 1, y + 1]
并且需要另一行代码来消除这种愚蠢。
coordinates = [cp for [cp] in coordinates]
我对Python的美丽非常着迷,并希望能够让我自己的代码变得更迷人一些。
谢谢!
酷列表理解:
def get_data_locations():
""" Find data locations from the first sheet in document. """
first_sheet = Range('Sheet1', 'A1:Z200').value
coordinates = []
for row_i, row in enumerate(first_sheet, 1):
coordinates.extend([(row_i, col_i) for col_i, col in enumerate(row, 1) if col == 'CO'])
return coordinates
非常感谢谁用这个解决方案提出了迈克·米勒!我重新发布了他的代码的一个稍微修改过的版本,以表明BBrown的建议的价值。拥有有意义的名字对于像我这样的初学者来说是一个不同的世界。
每行中是否只有一个单元格的值为“CO”? – inspectorG4dget
你可以发布一个实际运行的独立示例;与进口和所有。这可以让你更容易地尝试你的方法。 –
@ inspectorG4dget'CO'可以在一行中出现多次。 – Kristjan