pandasを使用したデータ処理に関して

Question

###前提・実現したいこと
以下のようなデータフレームtableAを作成したとします。
tableAの最初の一行のみを抜き出して、新しいデータフレームを作成したいです。

import pandas as pd
tableA = pd.DataFrame([["0001", "Smartphone"],
                      ["0001", "Wireless Charger"],
                      ["0002", "Wearable watch"]],
                      columns=['customer_id', 'product_name'])

いろいろな方法を試しましたが、上手くいきませんでした。
良い方法などありましたら、ご教授頂ければ幸いです。

よろしくお願いいたします。

###試したこと
iloc関数などを使って一行を抜き出して代入したりしましたが、上手くいきませんでした。
どうも戻り値がSeriesという方で帰ってきているのが問題のような気がします。

###補足情報(言語/FW/ツール等のバージョンなど)
Python3

Answer 1

Series ではなくて DataFrameで得たいのすね。
であれば

tableB = tableA.iloc[:1]

のような範囲指定でいけると思います。

ただし、上記の場合はコピーではなくてリンクとなりますので注意してください。（tablaA の値を変更すると tableB の値も変更されます）

それが用途と違うのであれば copy() を使って

tableC = tableA.iloc[:1].copy()

のようにしてください。
あと、下記のようにFilterを使ってもコピーになったかと思います。

tableD = tableA[tableA.index == 0]

Comments

[papi_tokei]

回答ありがとうございます．
回答いただいたとおり修正すると，適切に動作しました．
少しお聞きしたいのですが，tableA.iloc[1]でも一行取り出すことが可能だと思います．
私は上記の方法を使っていたのですが，コロンを入れることで，どのような動作になるのでしょうか？

[magichan]

Index指定をした場合、元データよりも次数の低いデータが得られますので、その型も次数に応じたものになります
（DataFrameはの場合はSeriesで、Seriesの場合はスカラー値で得られます）
ところが、範囲指定のスライスを行った場合は、（基本的に）元データと同じ次数のデータが得られますので、当然型も同じものになります。
ということで、今回はワザと範囲指定を使って１行だけ選択したということです。

[papi_tokei]

ありがとうございます．
アバウトですが理解できました．
スライスを使うとDataFrameがかえるので，今回はスライスで一行選択したということですね．
通常の場合は，1次元のベクトルデータが帰ってくるのだと理解しました．