如何将变量传递给pd.read_excel的'use_cols'参数

问题描述

我有如下代码

df = pd.read_excel(filepath,sheet_name=sheet_name,skiprows=skiprows,use_cols='A:O')

这很好用。但是，列在工作表之间变化，因此我想向用户提供input选项，在其中输入from_col和to_col的列名（A，B ..）变量，然后在use_cols参数中使用这些名称。

但是，我不能直接在参数use_cols中使用变量。我现在正在做的事情如下

from_col = 'A'
to_col='O'
a_l = string.ascii_uppercase
w_l=a_l[a_l.index(from_col):a_l.index(to_col)]
df = pd.read_excel(filepath,use_cols=w_l)

现在，问题是，有没有办法将变量直接传递给pd.read_excel的'use_cols'参数？还是比我现在使用的更简单的方法？

更新

上面使用的代码无法正常工作，无论我在O和from_col中传递了什么变量，它都读取到列to_col，不确定原因。列表w_l正确更新，但是use_cols似乎忽略了它！

解决方法

您可以简单地创建一个字符串并将其作为这样的参数传递：

from_col = 'A'
to_col='O'
w_l = f"{from_col}:{to_col}"  # 'A:O'
df = pd.read_excel(filepath,usecols=w_l)