テキストファイルからPythonのNumpyで3次元配列を作りたいです。

Question

カンマと改行で区切られたテキストファイルから3次元配列を作りたいです。
元のテキストファイルは以下の形になっており、あいうの部分は文章が入っています。

あ, い
う, え
お, か ……

これを以下のような形の配列にして読み込みたいです。

list = [[[あ],[い]]
        [[う],[え]]
		[[お],[か]]]

ファイルを読み込んだ後にsplit()でリストにしていたのですがうまくいかなかったので質問しました。
もしNumpyを使わずにできるのであればその手法でもかまいません。
よろしくお願いいたします。

Answer 1

NumPyを使わない方法だとこんな感じです。

data = []
with open('data.txt') as fin:
    for row in fin:
        row = row.rstrip()
        data.append([])   
        
        for elem in row.split(', '):
            data[-1].append([elem])

print(data)

実行結果 Wandbox

[[['あ'], ['い']], [['う'], ['え']], [['お'], ['か']]]

---

慣れてきたらこのようにも書けます。

with open('data.txt') as fin:
    data = [
        [[elem] for elem in row.split(', ')]
        for row in map(str.rstrip, fin)
    ]

print(data)

---

カンマと改行で区切られたテキストファイル

カンマの前後の無駄な空白が無ければ、csvモジュールを使えるのですが。
**追記: ** csvモジュールでも対応可能なそうです。can110さんありがとうございます。

import csv

data = []
with open('data.csv', newline='') as fin:
    for row in csv.reader(fin, skipinitialspace=True):
        data.append([])
        
        for elem in row:
            data[-1].append([elem])

print(data)

実行結果 Wandbox

[[['あ'], ['い']], [['う'], ['え']], [['お'], ['か']]]

Comments

[can110]

「 あ, い, う」などのようにカンマの後ろ（＝値の手前）だけに空白がついている場合はskipinitialspaceが使えますね。

[nuii_]

回答ありがとうございます。申し訳ありません、空白は手癖で入れてしまっていたものでテキストデータには入っていないものでした。
提案していただいた手法でも希望通りの配列ができそうでしたが、CSVモジュールでの手法もお聞きしてよろしいでしょうか？

[LouiS0616]

@can110 さん
そのようなオプションがあったのですね。ご教示ありがとうございます。

@nuii_ さん
追記しました。

[nuii_]

回答ありがとうございます。
csvモジュールで配列を作ることができました。

Answer 2

numpy.loadtxtで直接CSVを読み込んで.reshapeすると楽です。

import numpy as np
a = np.loadtxt('inp.csv',dtype=str,delimiter=',')
print(a)
"""
[['あ' 'い']
 ['う' 'え']
 ['お' 'か']]
"""

a = a.reshape(-1,2,1)
print(a.shape) # 3,2,1
print(a)
"""
[[['あ']
  ['い']]

 [['う']
  ['え']]

 [['お']
  ['か']]]
"""

# 確認
a = np.array(  [[['あ'],['い']],
                [['う'],['え']],
                [['お'],['か']]])
print(a.shape) # 3,2,1
print(a)
# 同じ

Comments

[nuii_]

回答ありがとうございます。申し訳ありません、空白は手癖で入れてしまっていたものでテキストデータには入っていないものでした。
提案していただいた手法で希望通りの配列が出来たのですが、手法のどの部分で空白を入れているのかお聞きしてもよろしいでしょうか？

[can110]

> 手法のどの部分で空白を入れているのか
この空白とは、回答内のどの部分のことを指していますか？
print(a)の結果の空行のことであれば、numpyが人間に見やすいように内部で勝手に入れているだけなので気にする必要はありません。

[nuii_]

回答ありがとうございます。
どの部分で空行が入ってるのか分からなかったのですっきりしました。