使用MultiIndex在pandas DataFrame中条件选择行

import pandas as pd

# 创建 MultiIndex
arrays = [
    ['2020', '2020', '2021', '2021'],
    ['Jan', 'Feb', 'Jan', 'Feb']
]
index = pd.MultiIndex.from_arrays(arrays, names=('Year', 'Month'))

# 创建 DataFrame
df = pd.DataFrame({'Sales': [100, 150, 200, 250]}, index=index)
print(df)

              Sales
Year Month          
2020 Jan         100
     Feb         150
2021 Jan         200
     Feb         250

# 选择 2020 年的数据
df_2020 = df.loc[2020]
print(df_2020)

# 选择 2020 年 1 月的数据
df_2020_jan = df.loc[(2020, 'Jan')]
print(df_2020_jan)

# 选择所有 1 月的数据
df_jan = df.xs('Jan', level='Month')
print(df_jan)

# 错误的索引值
try:
    print(df.loc[(2022, 'Jan')])
except KeyError as e:
    print(f"Error: {e}")

# 正确的做法是先检查索引是否存在
if (2022, 'Jan') in df.index:
    print(df.loc[(2022, 'Jan')])
else:
    print("指定的索引不存在")