2
3

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 3 years have passed since last update.

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

Last updated at Posted at 2020-02-07

#目的

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の内容が一次元の配列に格納されていることがわかる。

2
3
1

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
2
3

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?