NEW ARTICLE
MariaDB クエリの結果をcsvに出力する
  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]]