文字列から多(2)次元配列を作りたい

Question

前提

名前だけの文字列があります。
それを一次元配列に変更しています。

実現したいこと

20個ままでは一次元配列のままで良いのですが 20個以上 の場合は多次元配列にしたいです。

20個以下の場合
[name01, name02,...name20]

20個以上の場合
[
[name01, name02, ...name20],
[name21, name22, ...name40]
]

20個単位ごとに2次元配列に追加していきたいです。

条件

ユーザー名は100個までしかは入りません

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

二次元配列化できない

該当のソースコード

【現状】
// 初期化
var name: String = ""
// 20個以下の配列
var array: [String] = []
// 20個以上の配列
var twoDimensionalArray: [[String]] = [[]]

// 20個以下
if name.count <= 20 {
for i in 0..< self.name.count {
self.array.append(name)
print("----------------------------")
print("arrayは")
print(self.array)
print("----------------------------")
} 

// 20個以上
} else {
// ここで20個ずつの２次元配列にしたい
for i in 0..< self.name.count {
self.array.append(name)
}

// 回数
if( (self.array.count % 20) == 0 ){
// 何もしない
let count: Int = (self.array.count%20)
} else {
// +1する
let count: Int = (self.array.count%20)+1
}

// 21個以上40未満までのケース
if count == 2 {
let array01 = array[0..<20]
let array02 = array[21..<40]
twoDimensionalArray[0].append(array01)
twoDimensionalArray[1].append(array02)

// 41個以上60未満までのケース
} else if count == 3 {
let array01 = array[0..<20]
let array02 = array[21..<40]
let array03 = array[41..<60]
self.twoDimensionalArray[0].append(array01)
self.twoDimensionalArray[1].append(array02)
self.twoDimensionalArray[2].append(array03)

// 61個以上80未満までのケース
} else if count == 4 {
let array01 = array[0..<20]
let array02 = array[21..<40]
let array03 = array[41..<60]
let array04 = array[61..<80]
self.twoDimensionalArray[0].append(array01)
self.twoDimensionalArray[1].append(array02)
self.twoDimensionalArray[2].append(array03)
self.twoDimensionalArray[3].append(array04)
}

// 81個以上100未満までのケース
else if count == 5 {
let array01 = array[0..<20]
let array02 = array[21..<40]
let array03 = array[41..<60]
let array04 = array[61..<80]
let array05 = array[81..<100]
self.twoDimensionalArray[0].append(array01)
self.twoDimensionalArray[1].append(array02)
self.twoDimensionalArray[2].append(array03)
self.twoDimensionalArray[3].append(array04)
self.twoDimensionalArray[4].append(array05)
}

print("----------------------------------")
print("twoDimensionalArrayは")
print(twoDimensionalArray)
print("----------------------------------")
}

試したこと

・reduce
・append[添字]

配列から二次元配列へ20個ごとに区切って追加する方法が知りたいです。
初心者なのでお手柔らかにお願いいたします。

Comments

[hoshi-takanori]

Swift では、array という同じ変数を、ある時は1次元、またある時は2次元の配列として扱うことはできません。

[退会済みユーザー]

追記しました。
ご確認のほどよろしくお願いします。

[MasakiHori]

普通は20個以下でも2次元配列で扱う設計にしますね（[[name01,name02]]とかね）
なぜ20以下とそれを超える場合で分ける必要があるのかもう一度考えてみた方がいいです
それともそういう課題ですか？

[退会済みユーザー]

課題ではないです。

[退会済みユーザー]

20個以下でも以上でも二次元配列で扱う方向にしてみます。

Answer 1

こういう感じ

func f(_ array: [String], num: Int = 20) -> [[String]] {
    var copy = array[...]
    var result: [[String]] = []
    while copy.count > 0 {
        result.append(Array(copy.prefix(num)))
        copy = copy.dropFirst(num)
    }
    return result
}

で

let r = f(array)

if r.count < 2 {
    let ppp = r.flatMap { $0 }
    print(ppp)
} else {
    print(r)
}

こういうすると20以下の時は特別扱い。

Comments

[退会済みユーザー]

回答ありがとうございます。

No 'dropFirst' candidates produce the expected contextual result type '[String]' のエラーが出てしまいます。

```
private func f(_ array: [String], num: Int = 20) -> [[String]] {
var copy = self.array
var result: [[String]] = []
while copy.count > 0 {
result.append(Array(copy.prefix(num)))
copy = copy.dropFirst(num)
}
return result
}
```

[MasakiHori]

提示されたコード断片からは状況がわからなかったためメソッドではなく関数で書いています。
メソッドとして実装し self.array を利用するのであれば第一引数は不要です。

エラーに関してですが、私が書いた
var copy = array[...]
の [...] は飾りでも落書きでもありません。必要なものです。

[...] はArrayからそのすべての要素を持ったArraySliceを生成するものです。
dropFirst() はArrayではなくArraySliceを返すためあらかじめ変数copyの型をArraySliceにしています。

なお変数名や関数名は私が動作確認のために適当に付けたものなので付けなおすことをお勧めします。

[退会済みユーザー]

ありがとうございます。
無事解決しました。

こちらがちゃんとコード読まずに返信してご迷惑をお掛けしました。
教えていただいたコードでちゃんと解決できました！