csvファイルからnumpy配列に変換したい．

Question

前提・実現したいこと

x,y座標を記録しているcsvファイルを読み込んで，各座標を10倍にしてnumpy配列に変換したいです．

初学者のため，質問内容に語弊があるかもしれません...

該当のソースコード

import numpy as np
import csv

filename = 'data.csv'
with open(filename, 'r') as f:
reader = csv.reader(f)
data = [points for points in reader]

#data.csvの中身
"[102, 142]","[397, 124]"
"[210, 75]","[206, 274]","[245, 426]"
"[322, 194]","[297, 293]","[246, 373]","[162, 433]"

Answer 1

csvファイルの1~3段目ごとに2次元配列としてnumpy変換したいです

行ごとに要素数が異なりますので、numpy.ndarray のリストに変換するのはどうでしょうか。

import numpy as np
from ast import literal_eval

filename = 'data.csv'

with open(filename, 'r') as f:
    data = [np.array(literal_eval(l.replace('"', '')))*10 for l in f]

from pprint import pprint
pprint(data)

#
[array([[1020, 1420],
       [3970, 1240]]),
 array([[2100,  750],
       [2060, 2740],
       [2450, 4260]]),
 array([[3220, 1940],
       [2970, 2930],
       [2460, 3730],
       [1620, 4330]])]

以前の回答

ast.literal_eval を使ってもよいかと思います。

import numpy as np
from ast import literal_eval

filename = 'data.csv'

data = []
with open(filename, 'r') as f:
    for l in f:
        data.extend(literal_eval(l.replace('"', '')))

data = np.array(data) * 10
print(data)

#
[[1020 1420]
 [3970 1240]
 [2100  750]
 [2060 2740]
 [2450 4260]
 [3220 1940]
 [2970 2930]
 [2460 3730]
 [1620 4330]]

Comments

[c0720k]

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

質問の追記漏れで申し訳ないのですが，csvファイルの1~3段目ごとに2次元配列としてnumpy変換したいです．

[melian]

行ごとに要素数が異なりますので、そのままでは２次元配列にはできません。