如何根据Python中的列标题值选择excel列

Question

我有一个python数据框，我使用以下代码将其粘贴到excel工作表中：

df.to_excel（作家，列=[周刊，周刊1，周刊2]，sheet_name='QTY SLS'，启动=5，启动=8，头=假，索引=假）

在数据帧weeknum、Weeknum1和Weeknum2中选择的列是代码前面的输入（例如weeknum=第14周）。因此，这些可能是[‘第16周’、‘第15周’、‘第14周’]作为一个例子。我的问题是如何根据Weeknum输入选择开始列。因此，我粘贴到的Excel工作表如下所示：

所以，如果我把它放在输入中，我需要Startco的代码从第14周开始，如果我把它放在输入中，则从第10周开始。

希望这是有道理的。

Answer 1

你可以找到这样的开始列：

import pandas as pd

# Toy dataframe
df = pd.DataFrame(
    {
        "Week 1": [1, 1, 1],
        "Week 2": [2, 2, 2],
        "Week 3": [3, 7, 3],
        "Week 4": [4, 8, 4],
        "Week 5": [7, 2, 9],
    }
)

Weeknum = "Week 3"  # Position in the df header index == 2
Weeknum1 = "Week 4"
Weeknum2 = "Week 5"

# Get the position of df column with value of Weeknum
print(list(df.columns).index(Weeknum))
# Outputs 2

因此，您可以这样修改代码：

df.to_excel(
    writer,
    columns=[Weeknum, Weeknum1, Weeknum2],
    sheet_name="QTY SLS",
    startrow=5,
    startcol=list(df.columns).index(Weeknum),
    header=False,
    index=False,
)