numpyで読み込んだ文字列を分割したい

Question

実行環境はgoogle Colaboratoryです。

以下のコードでCSVを読み込んでいます。

testList = np.loadtxt('/content/drive/MyDrive/Colab Notebooks/test.csv',delimiter=',',dtype='unicode')
print(testList)

上記の出力は、
[["cat","A1"],
["dog","B1"],
["woif","B2"]]
となるのですが
[["cat","A","1"],
["dog","B","1"],
["woif","B","2"]]
にしたいと考えています。

スライスでtestList[:1][:0]とすると出力がなくなり、testList[:1][:1]とすると
["A1","B1","B2"]となり、スライスによって文字列を分割することができません。

どうすればいいでしょうか。

Comments

[1T2R3M4]

ペンとノートを用意して手順でも書いてみてはいかがでしょうか。

Answer 1

import numpy as np
import io

csv_data = '''
cat,A1
dog,B1
woif,B2
'''.strip()

testList = np.loadtxt(io.StringIO(csv_data), delimiter=',', dtype='unicode')
testList = np.apply_along_axis(
  lambda x: np.array([x[0]]+[*x[1]], dtype=object), -1, testList)

print(testList)

#
[['cat' 'A' '1']
 ['dog' 'B' '1']
 ['woif' 'B' '2']]

Answer 2

こういうことｗやりたい場合は、pandasを使った方が楽です。

import pandas as pd

df = pd.read_csv('/content/drive/MyDrive/Colab Notebooks/test.csv', header=None)
df['ABC'] = df[1].str[0]
df['num'] = df[1].str[1]
df.drop(1, axis=1, inplace=True)
testList = df.values
print(testList)

実行結果

>>> print(testList)
[['cat' 'A' '1']
 ['dog' 'B' '1']
 ['woif' 'B' '2']]

参考資料
pandasでcsv/tsvファイル読み込み（read_csv, read_table）
pandasで文字列にスライスを適用して任意の位置・長さの部分を抽出
pandasでcsvファイルの書き出し・追記（to_csv）