NEW ARTICLE
kotlin 正規表現を使用して文字列が部分一致で含まれているかを判定する
  1. HOME
  2. python
  3. python pandasで読み込んだCSVから配列を作成する

python pandasで読み込んだCSVから配列を作成する

python pandasで読み込んだCSVから配列を作成する

pythonで、ライブラリpandasのread_csvを使用して、読み込んだCSVから配列を作成するサンプルコードを記述してます。pythonのバージョンは3.8.5を使用してます。

目次

環境

  • OS windows10 pro 64bit
  • python 3.8.5

pandasインストール

pandasをインストールされていない方は、pipでインストールしておきます。

pip install pandas

# Successfully installed pandas-1.1.4

read_csvt使い方

read_csvを使用すると、読み込んだCSVから配列を作成することが可能です。

以下は「sample.csv」を読み込んでヘッダー情報とデータを配列化するサンプルコードとなります。

sample.csv

ソースコード

import pandas as pd

arr = pd.read_csv("sample.csv")

header = arr.columns.values
v = arr.values

print(header)
# ['name' 'age' 'tel']

print(v)
# [['aaa' 11 123456]
#  ['bbb' 22 123456]
#  ['ccc' 33 123456]]

Tプロパティを使用しても同一の結果を得られます。

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]]) 
print(arr.T)
# [[1 3 5]
#  [2 4 6]]