dfをsplitする

2024年6月8日

DataFrameの各セルの文字列を空白で分割し、それぞれの要素を新しい列に挿入する方法を以下に示します。

python
import pandas as pd

# 仮のDataFrameを作成
data = {'ITEM': ['item1 item2', 'item3 item4', 'item5 item6']}
df = pd.DataFrame(data)

# DataFrameを分割する関数
def split_dataframe(df):
    # 分割後のデータを保持するリストを初期化
    new_data = {'ITEM_1': [], 'ITEM_2': []}

    # DataFrameの各行に対して処理を行う
    for index, row in df.iterrows():
        # 空白で分割
        split_items = row['ITEM'].split()
        
        # 分割後の要素を新しい列に追加
        if len(split_items) >= 2:
            new_data['ITEM_1'].append(split_items[0])
            new_data['ITEM_2'].append(split_items[1])
        else:
            new_data['ITEM_1'].append('')
            new_data['ITEM_2'].append('')

    # 新しいDataFrameを作成
    new_df = pd.DataFrame(new_data)

    return new_df

# DataFrameを分割
new_df = split_dataframe(df)

print(new_df)

このコードでは、元のDataFrameを処理して新しいDataFrameを作成しています。新しいDataFrameには、元のDataFrameの各行の文字列が空白で分割され、分割後の要素が新しい列に挿入されます。