一次元配列を二次元配列に変換したい

Question

前提・実現したいこと

配列arrayが

[["100","150","140","120","130","110"],["2","4","3","5"]]

とあり、
この配列を

[[["100","140","130"],["150","120","110"]],[["2","3"],["4","5"]]

のように変形したい。

該当のソースコード

li = []
for i in range(len(array)):
  num += 1
  if num %2 ==0:
     #この先がわからない
 

Comments

[fuzzball]

元データの内側のキーは 0, 1 の二つで固定なのでしょうか？

[kaitotokai]

はい、固定です

[fuzzball]

なぜ辞書なのでしょう？配列/リストには出来ないのでしょうか？ {0:"100",1:"120"} → [100, 120] など。

[kaitotokai]

元のデータ型がそうだからです

[kaitotokai]

配列/リストには元のデータを変形すればできます

[fuzzball]

質問とは直接関係なく、ちょっと気になっただけですので‥。

Answer 1

#初期化部分はループの外に出す
if i not in dct:
    dct[i] = [[], []] #二次元配列で初期化

#追加
for key, value in d.items():
    dct[i][key].append(v[key]) #内側の配列に追加

こんな感じかな。雰囲気ぐらいは伝わるかとｗ
ダメだったら試行錯誤してみて下さい。

Comments

[fuzzball]

編集前の質問に対する回答です。念のため。

[kaitotokai]

答えてくださりありがとうございました！！

Answer 2

サードパーティモジュールmore-itertoolsを使うと楽で、しかも柔軟です。

from more_itertools import distribute

src = [["100", "150", "140", "120", "130", "110"], ["2", "4", "3", "5"]]
dst = [[list(e) for e in distribute(2, row)] for row in src]

print(dst)
# => [[['100', '140', '130'], ['150', '120', '110']], [['2', '3'], ['4', '5']]]

---

標準機能だけで書くなら、こんな感じでしょうか。

src = [["100", "150", "140", "120", "130", "110"], ["2", "4", "3", "5"]]
dst = [
    [row[0::2], row[1::2]] for row in src
]

print(dst)

ついでに

編集前の質問については、こんな感じで書けそう。

from collections import defaultdict

src = {
    "a1_0": {0: "100", 1: "120"}, "a1_1": {0: "150", 1: "130"}, "a1_2": {0: "140",1: "110"},
    "b1_0": {0: "2", 1: "3"}, "b1_1": {0: "4", 1: "5"}, 
    "c1_0": {0: "90", 1: "80"}, "c1_1": {0: "40", 1: "50"}
}

dst = defaultdict(list)
for key, value in src.items():
    dst[key[:2]].append(
        [value[k] for k in sorted(value)]
    )

for key, value in dst.items():
    dst[key] = list(zip(*value))

Comments

[kaitotokai]

ありがとうございます。ちなみに、編集前の質問に関してですが、"100"・"120"などの子辞書のバリューにnanがあった時、nanを1に置換するにはどうコードが書けますか？
dst[key[:2]].append([value[k] if value[k] is not 'nan' else 1 for k in sorted(value)])

[kaitotokai]

と書いたのですが、nanは結局置換されなくて。。。

[LouiS0616]

is not じゃなくて != で比較してみたらどうなります？

[kaitotokai]

!=でも変わらないです

[LouiS0616]

nanは間違いなく文字列ですか？
文字列だとしても、NaNだったりNanだったりしませんか？

[kaitotokai]

[nan, 1] のように入っているので、nanだとは思うのですが...

[LouiS0616]

type関数に放り込んで型を調べてみてください。

[kaitotokai]

nanは<class 'float'>と出ました。

[kaitotokai]

nanってfloatなんですか...?

[LouiS0616]

不定値を表すものですね。nanのほかにinfもあります。
条件部を value[k] is not np.nan あるいは not math.isnan(value[k]) にすれば良いかと思います。

[kaitotokai]

ありがとうございます。value[k] is not np.nanでできました！！