pandasを使用してfor文でセルの一部を抽出すると結果が全部同じ内容になる

python初心者です。お願いします。

下記のデータから「-」と「X」の間を数字を抽出するプログラムを書いていますが、
抽出結果が全て同じ内容になります。

df["品名"]
AAGGU-900X600BB
AAADG-2400X650BB
AAGH-2000X620BB
AAAGH-1800X600BB
AAAAAAAH-2200X650BB
AGH-2550X650BB
AAAAAAAAKV-2550X900BB
AAAAAAAKV-2245X650BB
AAASLKH-2500X650###

実際書いたプログラムが下記です。実行するとデータの最後の抽出結果「2500」がdf["幅"]の
全てのセルに入力されます。繰り返し処理が出来ているかdf["品名"]の内容をprintで確認しましたが
問題無く出力されます。

for T_Hinmei in df["品名"]:
f_position = T_Hinmei.find("-")
e_position = T_Hinmei.find("X")
df["幅"] = T_Hinmei[f_position+1:e_position]

初歩的な質問で本当に申し訳ありませんが、よろしくお願いいたします。

行動規範の内容に同意します

回答1件

ベストアンサー

実行するとデータの最後の抽出結果「2500」がdf["幅"]の全てのセルに入力されます。

df["幅"] = T_Hinmei[f_position+1:e_position]

df["幅"] としているので、列全体に同じ値が挿入されることになってしまいます。そのため、「幅」列の全ての値が「品名」列の最後の要素である AAASLKH-2500X650 の 2500 になっているわけです。
この様な場合は pandas.series.str.extract() を使うとよいかと思います。

python
1import pandas as pd
2
3df = pd.DataFrame({
4    "品名": ['AAGGU-900X600BB', 'AAADG-2400X650BB', 'AAGH-2000X620BB',
5             'AAAGH-1800X600BB', 'AAAAAAAH-2200X650BB', 'AGH-2550X650BB',
6             'AAAAAAAAKV-2550X900BB', 'AAAAAAAKV-2245X650BB', 'AAASLKH-2500X650']
7})
8
9df["幅"] = df["品名"].str.extract(r'-(.+?)X')
10print(df)
11
12#                     品名    幅
13# 0        AAGGU-900X600BB   900
14# 1       AAADG-2400X650BB  2400
15# 2        AAGH-2000X620BB  2000
16# 3       AAAGH-1800X600BB  1800
17# 4    AAAAAAAH-2200X650BB  2200
18# 5         AGH-2550X650BB  2550
19# 6  AAAAAAAAKV-2550X900BB  2550
20# 7   AAAAAAAKV-2245X650BB  2245
21# 8       AAASLKH-2500X650  2500