Luaでsplit関数を自作したがうまく動かない

Question

前提

Luaでsplit関数を自作しようと正規表現を使わない簡単な関数を作ってみましたが空白を返すだけでうまく動きません。

実現したいこと

• 第一引数を第二引数で分割したい
• • split("a-b-c","-","2") --bを返す

発生している問題・エラーメッセージ

空白を返す

該当のソースコード

function split(str,a,b)
	local after={}
	local text=""
	for i=1,#str do
		if (string.sub(str,i,i) == a) then
			after[i]=text
			text=""
		else
			text=text..string.sub(str,i,i)
		end
	end
	return after[tonumber(b)] --環境の都合上引数bは文字列で渡されます
end

試したこと

• bを数字にして実行

補足情報（FW/ツールのバージョンなど）

開発環境はMediaWikiのモジュール。

Comments

[ozwk]

for i=1,str do -- str ?

[Funa-enpitu]

# 付け忘れでした。
修正します

[Funa-enpitu]

因みにこれでも動きません

Answer 1

function split(str, sep, idx)
  local splitted = {}
  local pre = 0
  for i=1, #str do
    if (string.sub(str, i, i) == sep) then
      table.insert(splitted, string.sub(str, pre, i-1))
      pre = i + 1
    end
  end
  table.insert(splitted, string.sub(str, pre, #str))
  return splitted[tonumber(idx)]
end

print(split("a-b-c", "-", "2"))

# b

Comments

[Funa-enpitu]

動きました！ありがとうございます。insertを使うのですね

[Funa-enpitu]

あとですが、回答して頂いたコードを使用してもよろしいでしょうか。
ライセンスの関係でCC BY-SA 4.0 (https://creativecommons.org/licenses/by-sa/4.0/ )の下で公開されます。

[melian]

はい、ご自由にどうぞ。ただ、回答したコードは文字列から配列を作成しているので効率があまりよろしくありません。

[Funa-enpitu]

では、使わせていただきます。改めてありがとうございました。