如何以整数值作为列名对 pandas dataframe 进行切片

Question

我正在使用以年份作为列名的数据集。 有没有办法切片。

下面是示例。 我想用客户数 0 对 dataframe 进行切片。

df.loc[df.2020==0,:]这会产生错误。

Department      2016    2017    2018    2019    2020
Electronics     2030    1029    1839    893     20
Babyshop        2030    1029    1839    893     0
Grocery         2030    1029    1839    893     2
Hardware        2030    1029    1839    893     10

Answer 1

这是你要找的吗？

import pandas as pd
from io import StringIO

df = pd.read_csv(StringIO("""
Department      2016    2017    2018    2019    2020
Electronics     2030    1029    1839    893     20
Babyshop        2030    1029    1839    893     0
Grocery         2030    1029    1839    893     2
Hardware        2030    1029    1839    893     10
"""), sep='\s+')

print(df[df['2020'] == 0])

Output：

  Department  2016  2017  2018  2019  2020
1   Babyshop  2030  1029  1839   893     0

如果你想查找超过 1 年内有 0 个客户的行，你可以用这样的 or 条件来组合它们：

print(df[(df['2019'] == 0) | (df['2020'] == 0)])

这是一篇关于根据多个条件选择行的更多信息。