【Python】csvファイルを1次元配列のデータとして格納する方法

目的

sample.csv

a,b,c,d,e

上記のようなcsvファイルを下のような一次元配列に格納する。

['a','b','c','d','e']

機械学習によって分類したlabelデータの結果として保存したcsvを、またpythonプログラム内で扱うために作成したが、よりよい方法があればコメントにて教えていただきたいです。

方法

sample.csvとread_csv_flatten.pyが同じディレクトリにある場合、以下のコードを実行することでdataにcsvの内容が一次元の配列に格納される。
同じディレクトリ内に無い場合は、適当なpathを指定する必要がある。

read_csv_flatten.py

import csv
import numpy as np

with open("sample.csv") as fp:
    reader = csv.reader(fp)
    data = [ e for e in reader ]

data = np.array(data).reshape(-1)

#以下確認のための出力
print(data)

結果

['a' 'b' 'c' 'd' 'e']

以上のように、csvの内容が一次元の配列に格納されていることがわかる。