分割した後から1つ目と2つ目を新しいカラムにそれぞれ入れて、CSV出力

2024年6月8日

分割後の1つ目と2つ目を新しい列に入れ、その後DataFrameをCSVファイルに出力する方法を以下に示します。

python
import pandas as pd
import numpy as np

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

# "ITEM"列を空白で分割する関数
def split_item_by_space(item):
    if isinstance(item, str):
        return item.split()
    else:
        return [np.nan, np.nan]

# "ITEM"列を空白で分割し、新しい列に入れる
df[['ITEM_1', 'ITEM_2']] = pd.DataFrame(df['ITEM'].apply(split_item_by_space).tolist(), index=df.index)

# 分割後の"ITEM"列を削除
df.drop(columns=['ITEM'], inplace=True)

# CSVファイルに出力
df.to_csv('output.csv', index=False)

print(df)

このコードでは、まず"ITEM"列を空白で分割して、その結果を新しい列に入れています。その後、分割前の"ITEM"列は削除され、DataFrame全体がCSVファイルに出力されます。