pandas dfにおける行方向データの連結の仕方

Question

下のようなデータフレームがあったとして、このデータフレームに、食品ページごとに取得した拡張データを連結していきたいです。

names, id
かぼちゃ,10604
キャベツ,10606
グリーンボール,10606
枝豆,10601

例えば[100,200,300,400,True]でそれぞれ['カロリー', 'タンパク質', '脂質', '塩分', '太りやすいか']です
まだカラム名を作成していないのですが、これを上のようなデータフレームに連結するにはどうしたいいですか。

Comments

[meg_]

> これを上のようなデータフレームに連結するにはどうしたいいですか。
連結後のデータフレームのイメージがよく分かりません。

また、コードを掲載されないと回答が難しいかと思います。

Answer 1

どうにもイメージが曖昧だけど、こんな感じかな。

import pandas as pd
import io

txt = """
names,id
かぼちゃ,10604
キャベツ,10606
グリーンボール,10606
枝豆,10601
"""

# 画面出力時の日本語の桁揃え
pd.set_option('display.unicode.east_asian_width', True)

# namesとidのデータフレーム生成
df = pd.read_csv(io.StringIO(txt))
print(df, end="\n\n")

# 新たに追加する列名
new_columns = ['カロリー', 'タンパク質', '脂質', '塩分', '太りやすいか']

# 最初のデータフレームを基にして、追加する列名も含めたデータフレームを生成
df = pd.DataFrame(df, columns=df.columns.values.tolist() + new_columns)
print(df, end="\n\n")

# 拡張データを用意
dat = [100, 200, 300, 400, True]

# idを指定して拡張データをデータフレームに設定
df.loc[df['id'] == 10604, 'カロリー': '太りやすいか'] = dat
print(df)

            names     id
0        かぼちゃ  10604
1        キャベツ  10606
2  グリーンボール  10606
3            枝豆  10601

            names     id  カロリー  タンパク質  脂質  塩分  太りやすいか
0        かぼちゃ  10604       NaN         NaN   NaN   NaN           NaN
1        キャベツ  10606       NaN         NaN   NaN   NaN           NaN
2  グリーンボール  10606       NaN         NaN   NaN   NaN           NaN
3            枝豆  10601       NaN         NaN   NaN   NaN           NaN

            names     id  カロリー  タンパク質   脂質   塩分 太りやすいか
0        かぼちゃ  10604     100.0       200.0  300.0  400.0         True
1        キャベツ  10606       NaN         NaN    NaN    NaN          NaN
2  グリーンボール  10606       NaN         NaN    NaN    NaN          NaN
3            枝豆  10601       NaN         NaN    NaN    NaN          NaN