在Python中将csv文件导入矩阵/数组

时间:2015-05-06 12:05:02

标签: python csv import

我正在尝试将包含字符串和数据矩阵的大型csv文件导入到Python中的数组/矩阵中。在MATLAB中,我曾经加载文件并简单地将其分配给矩阵,但在Python中似乎有点棘手。有人可以提出建议吗?我是Python的新手。谢谢大家。

3 个答案:

答案 0 :(得分:7)

您可以使用内置的csv模块将数据加载到多维列表中:

import csv

with open('data.csv', 'rb') as f:
    reader = csv.reader(f)
    data_as_list = list(reader)

print data_as_list
# [['data1', 1],
#  ['data2', 2],
#  ['data3', 3]]

答案 1 :(得分:4)

您可以使用pandas。

import pandas as pd
df = pd.from_csv('filename.csv')

如果分隔符不是','您可以使用sep关键字更改默认值,例如:

df = pd.from_csv('filename.csv', sep='\')

您将获得一个带有强大分析功能的数据框。

答案 2 :(得分:0)

import numpy as np
with open("your_file.csv",'r')as f :
    data_list = list(csv.reader(f, delimiter=";"))
data_array=np.array(data_list[1:])