pandas: loc indexer not return the correct type of object

[RuixiangLiu01]

Hi, developer. I meet a problem when using 'loc' to select some columns, it seems 'loc' doesn't return the a DataFrame for me

any advise would be appreciated, thanks

[RuixiangLiu01]

update:
I checked the stub files

I think my code should be inferred as a DataFrame not a Seires accroding to the code in stub file

[judej]

@RuixiangLiu01, could you please add a code snippet for this so that we could repro?

[RuixiangLiu01]

@RuixiangLiu01, could you please add a code snippet for this so that we could repro?

Thanks for reply. Here is a simple example:
df = pd.DataFrame({'A': [1, 2, 3, 4], 'B': [2, 3, 4, 5]}) df1 = df.loc[:,'A']

I am not sure is it helpful for u

[gramster]

This is a non-trivial fix; right now things only work as well as they do because of a bug :-( There's a bunch of complexity around this kind of indexing that I need to think through.

[jakebailey]

This issue has been fixed in version 2021.5.0, which we've just released. You can find the changelog here: https://github.com/microsoft/pylance-release/blob/main/CHANGELOG.md#202150-5-may-2021