pythonでdataframeから取り出したデータの型

Question

こちらのコードを実行したとき、
半角の二重カッコにしている箇所を意味を、どう理解すればよいかご教示いただけますと幸いです。

◆コード
ーーーーーーーーー
import pandas as pd
df = pd.DataFrame([list('abcd'),list('efgh'),list('ijkl')],columns=list('ABCD'))
df[['A']]
ーーーーーーーーー
結果はdataframe型が返ってきました。
df['A']（＝半角の１重カッコの場合）ではseriesが返ってくると思うのですが、そのseries型をdataframeへ変換している、と捉えればよいのでしょうか？

それともdataframe自体の機能として、二重カッコではdataframe型を返すと決まっている、と捉えるほうが適切でしょうか？
※jupyterで実行しております。

基本的なところで恐縮なのですが、ご教授いただけますと幸いです。
どうぞ宜しくお願い致します。

Answer 1

DataFrameの[]演算子で Columnを指定する場合、

• Column名を直接指定する方法
• column名のListで指定する方法

があり、前者の場合Series型のViewが、後者の場合はDataFrame型のViewが結果として得られます。

こんな感じ

import pandas as pd
df = pd.DataFrame([list('abcd'),list('efgh'),list('ijkl')],columns=list('ABCD'))

print(df['B'])
# 0    b
# 1    f
# 2    j
# Name: B, dtype: object

print(df[['B', 'C']])
#    B  C
# 0  b  c
# 1  f  g
# 2  j  k

で、今回の

df[['A']]

の解釈ですが、上記の後者の呼び方を使って、1つのColumnだけを指定しているということになります。当然結果は DataFrame型になります。

Comments

[nterka]

理解できました。ありがとうございます。
※サンプルコード付きで感謝いたします

Answer 2

二重カッコに特別な意味があるのではなく、「要素を一つだけ含むlistを引数に使っている」ということではないでしょうか。
df['A']のように単値('A')を渡すと次元の縮約が起こるのに対して、リストを渡すとそうならないということだと思います。

Comments

[nterka]

ありがとうございます。そのように理解したいと思います。