【Python】DataFrameの行分割についてのご質問

Question

「名称」「カテゴリ１」「カテゴリ２」の３カラムを持ったデータフレームがあります。
「カテゴリ２」のカラムには「002、003、004」の値を「002~004」と、まとめて表記されてしまったデータとなっています。

名称 カテゴリ１ カテゴリ２
A A-1 001
A A-1 002
B B-1 001
B B-1 002~004

質問になります。
上記、カテゴリ２の「002~004」を「名称」「カテゴリ１」の値を持ったまま、
以下のように別の行に分割する方法をご教示いただけませんでしょうか。

名称 カテゴリ１ カテゴリ２
A A-1 001
A A-1 002
B B-1 001
B B-1 002 ★
B B-1 003 ★
B B-1 004 ★

Answer 1

import pandas as pd
import io
import re

txt = """
名称,カテゴリ１,カテゴリ２
A,A-1,001
A,A-1,002
B,B-1,001
B,B-1,002~004
"""


def make_list(s):
    if re.fullmatch(r'\d{3}', s):
        return [s]
    if re.fullmatch(r'\d{3}~\d{3}', s):
        return [f"{i:03}" for i in range(int(s[0:3]), int(s[4:7]) + 1)]
    return None


df = pd.read_csv(io.StringIO(txt))
df['カテゴリ２'] = df['カテゴリ２'].apply(make_list)
# print(df)

df = df.set_index(['名称', 'カテゴリ１'])['カテゴリ２'].apply(pd.Series).stack().reset_index()
df.drop(columns=df.columns[2], inplace=True)
df.columns = ['名称', 'カテゴリ１', 'カテゴリ２']
print(df)

  名称 カテゴリ１ カテゴリ２
0  A   A-1   001
1  A   A-1   002
2  B   B-1   001
3  B   B-1   002
4  B   B-1   003
5  B   B-1   004

Comments

[iios]

ありがとうございます。
とても助かりました。

Answer 2

print(df)
#	名称	カテゴリ１	カテゴリ２
#0	A	A-1	001
#1	A	A-1	002
#2	B	B-1	001
#3	B	B-1	002~004

def f(x):
    l = x.split('~')
    temp = []
    for i in range(int(l[0]), int(l[-1]) + 1):
        temp.append(f'{i}'.zfill(3))
    return temp

df['カテゴリ２'] = df['カテゴリ２'].apply(f)
df = df.explode('カテゴリ２')
print(df)
#	名称	カテゴリ１	カテゴリ２
#0	A	A-1	001
#1	A	A-1	002
#2	B	B-1	001
#3	B	B-1	002
#3	B	B-1	003
#3	B	B-1	004

Comments

[iios]

ありがとうございます。
実データの方でもうまく分割されました。
とても助かりました。