正規表現で抜き出した数字をpandasのDataFrameに入れられない

Question

正規表現などを用いて抜き出した文字列をdataframeに保存しようと試したところ、なかなかうまくいきません。

実際のコードは以下のものになります

import pandas as pd
import csv
import re

lst = []
for i in range(1,5):
    url1 = 'https://kabuoji3.com/stock/?page={}'.format(i)
    print(url1)
    dfs = pd.read_html(url1)
    lst.extend(dfs)
    dfr = pd.concat(lst, ignore_index=True)

for j in dfr["コード・名称"]:
    code = re.search('[0-9]{4}' , j)
    print(code.group(),j[5:])
    dfr.assign("コード"=code.group())
    dfr.assign("銘柄名"=j[5:])

最後の2桁が想定通りの動きをしません。

ちなみにコードは4桁の数字、銘柄名は　（株）極洋
といったデータです。
この　コード　、銘柄名　のデータが1つになってしまっているため
分割して保存しようとしております。ご回答よろしくお願いします＞

Answer 1

先に列を追加しておき、ループではインデックスを指定して値をセットしてやればよいです。

import pandas as pd
import re
pd.set_option('display.unicode.east_asian_width', True)

dfr = pd.DataFrame({'コード・名称':['1301 (株)極洋', '1305 ダイワ 上場投信-トピックス']})

dfr = dfr.assign(コード='')
dfr = dfr.assign(銘柄名='')
for i,v in dfr["コード・名称"].items():
    code = re.search('[0-9]{4}' , v)
    print(code.group(),v[5:])
    dfr['コード'][i]=code.group()
    dfr['銘柄名'][i]=v[5:]

print(dfr)
#                      コード・名称 コード                      銘柄名
#0                    1301 (株)極洋   1301                    (株)極洋
#1  1305 ダイワ 上場投信-トピックス   1305  ダイワ 上場投信-トピックス

Comments

[TheBullchannel]

ありがとうございました！
教えていただいたコードを参考にして自分でも動かすことができました！
データフレームをループ外で作っておけばよかったのと、代入の仕方に問題があったのですね