ヒットしたデータ数によってPandasの`.loc`の返却値の型が異なる

問題

.loc[x]のxにiterableでない値を指定した場合、ヒットした行数によって返却値の型が異なる。
具体的にはヒットした行数が1の場合はpd.Seriesを、それ以外の場合はpd.DataFrameを返却する。

解決方法

同じ型で欲しい場合はxをiterableにして指定する

df = pd.DataFrame({"index": [1, 2, 2], "value": [4, 5, 6], "v2": [1, 1, 1]}).set_index(
    "index"
)
print(df)
print(type(df.loc[1]))
print(type(df.loc[2]))
print(type(df.loc[[1]]))
print(type(df.loc[[2]]))

       value  v2
index           
1          4   1
2          5   1
2          6   1
<class 'pandas.core.series.Series'>
<class 'pandas.core.frame.DataFrame'>
<class 'pandas.core.frame.DataFrame'>
<class 'pandas.core.frame.DataFrame'>