Pandas DataFrame mypy 错误：切片索引必须是整数或 None

Question

以下行

pd.DataFrame({"col1": [1.1, 2.2]}, index=[3.3, 4.4])[2.5:3.5]

引发 mypy linting 错误[2.5

切片索引必须是整数或 None

这是有效的语法并正确返回

     col1
3.3   1.1

如果没有# type: ignore，我该如何解决这个 linting 错误？

版本：

• 熊猫1.3.0
• 迈比 0.931

有问题的代码：

def get_dataframe(
    ts_data: GroupTs,
    ts_group_name: str,
    start_time: Optional[float] = None,
    end_time: Optional[float] = None,
) -> pd.DataFrame:
    df = pd.DataFrame(ts_data.group[ts_group_name].ts_dict)[
        start_time:end_time
    ].interpolate(
        method="index", limit_area="inside"
    )  # type: pd.DataFrame
    return df[~df.index.duplicated()]

Answer 1

我担心这是目前的设计，但如果必须的话，您可以通过使用可调用的切片来使 mypy 静音，如下所示：

import pandas as pd

df = pd.DataFrame({"col1": [1.1, 2.2]}, index=[3.3, 4.4])[
    lambda x: (2.5 <= x.index) & (x.index < 3.5)
]

print(df)
# Ouput
     col1
3.3   1.1

mypy 报告no issues found了这段代码。