Pandasのデータフレームを転置した後にindexを付けたい

Question

実現したいこと

  項目名   検索条件
0  名前   日本 花子
1  性別     女性

上のデータフレームを転置して、下のようなデータフレームを得るプログラムを書こうとしています。

    項目名     名前    性別
0  検索条件  日本 花子  女性

発生している問題

過去の質問を参考に、次のようなプログラムを書きました。

import pandas as pd
import io

csv = """
項目名,検索条件
名前,日本 花子
性別,女性
"""

df1 = pd.read_csv(io.StringIO(csv))
print(df1)

df2 = df1.T.reset_index()
print(df2)

しかしこれを実行すると、df2は次のようなデータフレームになってしまいます。

  index      0   1
0   項目名     名前  性別
1  検索条件  日本 花子  女性

試したこと

reset_index()する前に、df2 = df1.T.set_index("項目名")としてindexを指定しようとしたら、「KeyError: "None of ['項目名'] are in the columns"」というエラーになりました。自分なりに調べたのですが、類似のトラブルが見つからず質問させていただきました。環境はAnacondaを使っています。

Answer 1

以下のように、転置した結果の先頭行をカラム名、２行目以降をデータとしてデータフレームを作成すればよいかと思います。

import io

csv = """
項目名,検索条件
名前,日本 花子
性別,女性
"""

pd.set_option('display.unicode.east_asian_width', True)

df1 = pd.read_csv(io.StringIO(csv))

a = df1.T.reset_index().values
df2 = pd.DataFrame(a[1:,:], columns=a[0,:])
print(df2)
#     項目名       名前  性別
#0  検索条件  日本 花子  女性

Comments

[Sunset_Yuhi]

valuesでNumPy形式のリストを得て、データフレームに直すのは分かりやすいですね。ありがとうございます。

Answer 2

スマートかどうかわかりませんが、こんな感じでしょうか。

import pandas as pd
import io

csv = """
項目名,検索条件
名前,日本 花子
性別,女性
"""

df1 = pd.read_csv(io.StringIO(csv), header=None, index_col=0)
df2 = df1.T.reset_index(drop=True)
df2.columns.name = ""
print(df2)

ヘッダなしで読んでおかないと面倒です。

Comments

[Sunset_Yuhi]

説明不足で申し訳なかったのですが、読み込み方法を変えて対応するのは無しで、

項目名 検索条件
0 名前 日本 花子
1 性別 女性

というデータフレームを、

項目名 名前 性別
0 検索条件 日本 花子 女性

に変換するプログラムを書きたい、というのが今回の質問の意図でした。

Answer 3

過去の質問と同様に

df2 = pd.read_csv(io.StringIO(csv), index_col='項目名').T

で良いのではないでしょうか

Comments

[hayataka2049]

たぶん「項目名」列を作りたいのだと思います。

[Sunset_Yuhi]

質問が分かりにくくてすみません。ご指摘の通り、「項目名」の列を作りたいと考えています。